选择问题--模拟快排

选择问题

给定n个元素和一个整数k(1 <= k <= n),找出这n个元素中的第k小的元素

  • 可以对元素进行一次排序,然后遍历一遍便可以找到第k小的元素。时间复杂度为O(nlgn)。
  • 此处模拟快排,借助快排的思想,对数组递归的进行划分,与快排不同的是:只对划分出的子数组之一进行递归处理

将数组A[p,r]划分成A[p,q]和A[q+1,r],是的A[p,q]中的每个元素都不大于A[q+1,r]中的每个元素
接着计算子数组A[p, q]中元素的个数m:

  • 如果k<=m,则A[p,r]中第k小的元素落在子数组A[p,q]中
  • 如果k>m,则要找的第k小的元素落在子数组A[q+1,r]中。由于此时已经知道了A[p,q]中所有的元素都要小于第k小的元素,所以只需要在A[q+1,r]中找第k-m小的元素即可
int randomSelect(int *a, int p, int r, int k) {
	if (p == r)
		return a[p];
	int i = randomPartition(a, p, r);
	int m = i - p + 1;
	if (k <= m)
		return randomSelect(a, p, i, k);
	else
		return randomSelect(a, i + 1, r, k - m);
}

int randomPartition(int *a, int p, int r) {
	int i = p + rand() % (r - p + 1);
	swap(a[i], a[r]);
	return partition(a, p, r);
}

int partition(int *a, int p, int r) {
	int x = a[r];
	int i = p - 1;

	for (int j = p; j < r; j++) {
		if (a[j] <= x) {
			swap(a[++i], a[j]);
		}
	}
	swap(a[i + 1], a[r]);
	return i + 1;
}

时间复杂度:
最坏情况下:O(n^2),在找最小元素时,总是在最大元素出划分
平均时间复杂度:O(n)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值