lintcode在数组中找到第k大的元素(快速排序)

快速排序思想:先在数组中选择一个数,再把数组中的数字分成两部分,比选择数字小的放在左边,比选择数字大的放在右边。
这里写图片描述
代码:取数组最右边的数字作为选择的数字,从数组的第一个数字开始与选择数字进行比较,将小于选择数字放入数组左边,当挑选出所有小于选择数子的数时,将选择数字放入中间。则数组剩余的右边部分为大于等于选择数字的所有数。

int partition(int data[],int length,int start,int end){
	if(data==NULL||length<=0||start<0||end>=length)
		throw new std::exceotion("invalid parameters");
	
	int small=start-1;
	for(int i=start;i<end;i++){
		if(data[i]<data[end]){
		    if(small!=i)
			    std::swap(data[small],data[i]);
			small++;
		}
	}
	small++;
	std::swap(data[small],data[end]);
	return small;
}

small表示小于选择数字的下标,index表示遍历数组的小标。
这里写图片描述
data[1]与选择数字比较,若小于选择数字,则small向后移,index向后移。继续比较data[2]和选择数字。
这里写图片描述
data[2]与被选择数字比较,不小于选择数字,则index后移,small不移动。
这里写图片描述
当比较到第n个数字data[n]时,小于选择数字,将data[n]与data[small]交换,将小于选择的数字移动到数组左边。

1.i表示数组下标,依次取数组与被选择数字(数组最后一个数字)进行比较。
2.如果数组中的数字不小于被选择数字,则继续遍历下一个数字,i+1;
3.如果该数字小于被选择数字,small下标加1,表示小于被选择数组的部分多了一个。(small下标记录小于被选择数字的所有数字个数,即下表从0到small-1范围内的数小于被选择数。)
4.当下标small不等于遍历到的第i个数字下标i时,为了将小于被选择数字的所有数字都放在数组左边,交换data[small]和data[i]
5.最后交换数组被选择数字data[end]和data[small],将被选择数字放在数组中间,小于部分在右侧,大于等于部分在左侧。
这里写图片描述

这里写图片描述
递归的思想在每次选中的数字左右两边排序(递归:调用自身)
分别将得到所有小于选择数字的部分进行排序(start——index-1),将所有大于选择数字的部分进行排序(index+1——end)。(选择数字比左半部分大,比右半部分小所以递归时不需要将它进行排序)
代码:

void quicksort(int data[],int length,int start,int end){
	if(start==end)
		return;
	int index=partition(data,length,start,end);
	if(index>start)
		quicksort(data,length,start,index-1);//小于选择数字的部分进行排序
	if(index<end)
		quicksort(data,length,index+1,end);//大于选择数字的部分进行排序
}

在数组中找到第k大的元素
以某一个元素为pivot元素,将元素分为两个集合,一个集合元素比pivot小,另一个比pivot大
1.若比pivot大的元素数目正好为k-1,那么pivot就是我们要找到元素
2.若比pivot大的元素为m(小于k), 那么就在比pivot小的集合里面找第(k-m)大的元素
3.若是比pivot大的元素为m(大于k),那就继续在该集合里面找第k大的元素

 int findKthEle(vector<int>& nums, int start, int end,int k)
    {
        int index = partition(nums,start,end);
        int length = end-index+1;
        if(length == k)
            return nums[index];
        else if(length > k)
            return findKthEle(nums,index+1,end,k);
        else
            return findKthEle(nums,start,index-1,k-length);
    }
  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值