AcWing
  • 首页
  • 课程
  • 题库
  • 更多
    • 竞赛
    • 题解
    • 分享
    • 问答
    • 应用
    • 校园
  • 关闭
    历史记录
    清除记录
    猜你想搜
    AcWing热点
  • App
  • 登录/注册

topK问题的边界值选择

作者: 作者的头像   frankZ ,  2025-07-05 16:04:41 · 陕西 ,  所有人可见 ,  阅读 1


0


以i - 1为分界点

i - 1左边都是大于等于x
i右边都是小于等于x
并且x不能选择nums[l] 或者nums[l + r >> 1] 因为可能出在递归时出现死循环top(nums, l, r, k)

class Solution {
    public int findKthLargest(int[] nums, int k) {
        return topK(nums, 0, nums.length - 1, k);
    }

    private int topK(int[] nums, int l, int r, int k) {
        if (l >= r) {
            return nums[l];
        }

        int i = l - 1, j = r + 1;
        int x = nums[l + r + 1 >> 1];
        while (i < j) {
            do {i++;} while (nums[i] > x);
            do {j--;} while (nums[j] < x);
            if (i < j) {
                swap(nums, i, j);
            }
        }
        if (i - 1 -  l + 1 >= k) {
            return topK(nums, l, i - 1, k);
        } else {
            return topK(nums, i, r, k - (i - 1 - l + 1));
        }
    }

    private void swap(int[] nums, int l, int r) {
        int t = nums[l];
        nums[l] = nums[r];
        nums[r] = t;
    }
}

以j为分界点

j左边都大于等于x
j + 1右边都小于等于x
并且x不能选择nums[r] 或者nums[l + r + 1 >> 1] 因为可能出在递归时出现死循环top(nums, l, r, k)

class Solution {
    public int findKthLargest(int[] nums, int k) {
        return topK(nums, 0, nums.length - 1, k);
    }

    private int topK(int[] nums, int l, int r, int k) {
        if (l >= r) {
            return nums[l];
        }

        int i = l - 1, j = r + 1;
        int x = nums[l + r >> 1];
        while (i < j) {
            do {i++;} while (nums[i] > x);
            do {j--;} while (nums[j] < x);
            if (i < j) {
                swap(nums, i, j);
            }
        }
        if (j - l >= k) {
            return topK(nums, l, j, k);
        } else {
            return topK(nums, j + 1, r, k - (j - l + 1));
        }
    }

    private void swap(int[] nums, int l, int r) {
        int t = nums[l];
        nums[l] = nums[r];
        nums[r] = t;
    }
}

0 评论

App 内打开
你确定删除吗?
1024
x

© 2018-2025 AcWing 版权所有  |  京ICP备2021015969号-2
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息