寻找第K大(有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。 给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数。)

26 篇文章 3 订阅

寻找第K大

有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。  给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数。

测试样例:

[1,3,5,2,2],5,3
返回:2

分析:

根据快速排序的思路:

快排参见:https://blog.csdn.net/qq_43109561/article/details/90235946

按如下步骤在n个元素里面第K大的元素:

第一步:进行一次快排(将大的元素放在前半段,小的元素放在后半段),得到的中轴p
第二步:判断 p - low + 1 == k ,如果成立,直接返回a[p],(前半段有p - low(等于k-1)个大于a[p]元素,则a[p]为第K大的元素)
第三步:如果 p - low + 1 > k,则第k大的元 素在前半段,更新high = p - 1,执行第一步
第四步:如果 p - low + 1 < k,则第k大的元素在后半段,更新low = p + 1, 且 k = k - (p - low + 1),执行第一步

常规快排得到整体有序的数组复杂度是O(nlgn),此方法每次可以去掉“一半”的元素,复杂度为O(n)。

代码:

public class Finder {
    public int findKth(int[] a, int n, int K) {
        return quickSort(a,0,n-1,K);
    }
    private int quickSort(int[] arr, int low, int high,int K){
        int p = partion(arr,low,high);
        if(K == p-low+1){
            return arr[p];
        }else if(p-low+1 > K){
            //递归左边
            return quickSort(arr,low,p-1,K);
        }else{
            //递归右边
            return quickSort(arr,p+1,high,K-(p-low+1));
        }
    }
    private int partion(int[] arr, int low, int high){
        int tmp = arr[low];
        while(low < high){
            while(low < high && arr[high] <= tmp){
                high--;
            }
            if(low == high){
                break;
            }else{
                arr[low] = arr[high];
            }
            while(low < high && arr[low] >= tmp){
                low++;
            }
            if(low == high){
                break;
            }else{
                arr[high] = arr[low];
            }
        }
        arr[low] = tmp;
        return low;
    }
}

 

  • 7
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要求在O(n)的时间复杂度下找出整数数组的第k小元素,可以使用快速选择算法(QuickSelect)来实现。快速选择算法是基于快速排序算法的一种改进,它通过每次选择一个基准元素将数组分为两部分,并根据基准元素的位置来确定第k小元素所在的区间,从而减少了排序的时间复杂度。 以下是使用Java实现快速选择算法找出数组的第k小元素的代码: ```java import java.util.Random; public class QuickSelect { public static int findKthSmallest(int[] nums, int k) { if (nums == null || nums.length == 0 || k <= 0 || k > nums.length { throw new IllegalArgumentException("Invalid input"); } return quickSelect(nums, 0, nums.length - 1, k - 1); } private static int quickSelect(int[] nums, int start, int end, int k) { if (start == end) { return nums[start]; } int pivotIndex = partition(nums, start, end); if (pivotIndex == k) { return nums[pivotIndex]; } else if (pivotIndex < k) { return quickSelect(nums, pivotIndex + 1, end, k); } else { return quickSelect(nums, start, pivotIndex - 1, k); } } private static int partition(int[] nums, int start, int end) { int pivotIndex = new Random().nextInt(end - start + 1) + start; int pivot = nums[pivotIndex]; swap(nums, pivotIndex, end); int i = start; for (int j = start; j < end; j++) { if (nums[j] < pivot) { swap(nums, i, j); i++; } } swap(nums, i, end); return i; } private static void swap(int[] nums, int i, int j) { int temp = nums[i]; nums[i] = nums[j]; nums[j] = temp; } } ``` 使用示例: ```java public class Main { public static void main(String[] args) { int[] nums = {3, 2, 1, 5, 6, 4}; int k = 2; int result = QuickSelect.findKthSmallest(nums, k); System.out.println("第" + k + "小元素是:" + result); } } ``` 输出结果为: ``` 第2小元素是:2 ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值