hihocoder#1133之非有序数组二分查找k小数

最新推荐文章于 2019-09-16 21:16:23 发布

董道不渝

最新推荐文章于 2019-09-16 21:16:23 发布

阅读量641

点赞数

分类专栏： hihocoder 文章标签： hihocoder 二分查找

本文链接：https://blog.csdn.net/Q1204265228/article/details/44702909

版权

hihocoder 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

提示：非有序数组的二分查找之二

输入

第1行：2个整数N,k，N表示数组长度。

第2行：N个整数，表示a[1..N]，保证不会出现重复的数，1≤a[i]≤2,000,000,000。

输出

第1行：一个整数t，表示t在数组中是第k小的数，若K不在数组中，输出-1。

样例输入

10 4
1732 4176 2602 6176 1303 6207 3125 1 1011 6600

样例输出

#include
    
    
     
     
#include
     
     
      
      

using namespace std;

#define swap(x, t, y) (t = x, x = y, y = t)

int a[1111111], k;

int find(int s, int e)
{
	int i = s - 1, j, t, r, x = a[e];
	for(j = s; j < e; j++)
	{
		if(a[j] <= x)
		{
			i++;
			swap(a[i], t, a[j]);
		}
	}
	swap(a[i + 1], t, a[e]);
	if(i + 1 < k)
	{
		r = rand()%(e - i - 1) + i + 2;
		swap(a[r], t, a[e]);
		find(i + 2, e);
	}
	else if(i + 1 > k)
	{
		r = rand()%(i + 1 - s) + s;
		swap(a[r], t, a[i]);
		find(s, i);
	}
	else
		return i + 1;
}

int main(void)
{
	int i, n, r, t;
	cin>>n>>k;
	for(i = 1; i <= n; i++)
		cin>>a[i];
	if(k <= 0 || k > n)
		cout<<"-1\n";
	else
	{
		r = rand()%n + 1;
		swap(a[r], t, a[n]);
		cout<