您的位置:首页 > 其它

kth largest element in an array

2016-07-22 10:34 513 查看
找第k大,exch+partition+select:

private void exch(int[] a, int i, int j) {
int t = a[i];
a[i] = a[j];
a[j] = t;
}

private int partition(int[] a, int lo, int hi) {
int i = lo, j = hi + 1;
int v = a[lo];
while (true) {
while (a[++i] > v && i != hi) ;
while (a[--j] < v && j != lo) ;
if (i >= j) break;
exch(a, i, j);
}
exch(a, lo, j);
return j;
}

public int findKthLargest(int[] nums, int k) {
int lo = 0, hi = nums.length - 1;
while (lo < hi) {
int j = partition(nums, lo, hi);
if (j + 1 == k) {
return nums[k - 1];
} else if (j + 1 > k) {
hi--;
} else {
lo++;
}
}
return nums[k - 1];
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: