#1133 : 二分·二分查找之k小数 （ 快速排序， 分治 OR nth_element() 函数）

#1133 : 二分·二分查找之k小数

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

在上一回里我们知道Nettle在玩《艦これ》，Nettle的镇守府有很多船位，但船位再多也是有限的。Nettle通过捞船又出了一艘稀有的船，但是已有的N(1≤N≤1,000,000)个船位都已经有船了。所以Nettle不得不把其中一艘船拆掉来让位给新的船。Nettle思考了很久，决定随机选择一个k，然后拆掉稀有度第k小的船。 已知每一艘船都有自己的稀有度，Nettle现在把所有船的稀有度值告诉你，希望你能帮他找出目标船。

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

输入

第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

样例输出

1732

//快排分治，选择一半排序 31ms
#include<cstdio>
#include<algorithm>
//#include<bits/stdc++.h>
using namespace std;
template<class T>inline T read(T&x)
{
char c;
while((c=getchar())<=32)if(c==EOF)return 0;
bool ok=false;
if(c=='-')ok=true,c=getchar();
for(x=0; c>32; c=getchar())
x=x*10+c-'0';
if(ok)x=-x;
return 1;
}
template<class T> inline T read_(T&x,T&y)
{
return read(x)&&read(y);
}
template<class T> inline T read__(T&x,T&y,T&z)
{
return read(x)&&read(y)&&read(z);
}
template<class T> inline void write(T x)
{
if(x<0)putchar('-'),x=-x;
if(x<10)putchar(x+'0');
else write(x/10),putchar(x%10+'0');
}
template<class T>inline void writeln(T x)
{
write(x);
putchar('\n');
}
//-------ZCC IO template------
const int maxn=1000011;
const double inf=999999999;
#define lson (rt<<1),L,M
#define rson (rt<<1|1),M+1,R
#define M ((L+R)>>1)
#define For(i,t,n) for(int i=(t);i<(n);i++)
typedef long long  LL;
typedef double DB;
typedef pair<int,int> P;
#define bug printf("---\n");
#define mod 10007
LL a[maxn];
LL k;
void my_sort(LL left,LL right)
{

// for(int i=left;i<right;i++)
//  printf("%d ",a[i] );
// printf("\n");

if(left<right)
{
LL key=a[left];
LL low=left;
LL high=right;
while(low<high)
{
while(low<high&&a[high]>=key)
high--;
if(low<high)swap(a[high],a[low]);
while(low<high&&a[low]<=key)
low++;
if(low<high)swap(a[high],a[low]);
}
//  printf("[k==%lld , left =%lld  right=%lld ]\n",k,low,high);
if(low==k)return ;//选择一半进行排序即可
if(k<low)
my_sort(left,low-1);
else
my_sort(low+1,right);
}
}
int main()
{
//#ifndef ONLINE_JUDGE
//freopen("in.txt","r",stdin);
//#endif // ONLINE_JUDGE
LL n,m,i,j,t;
while(read_(n,k))
{
For(i,1,n+1)
{
read(a[i]);
}
// nth_element(a,a+k-1,a+n);
my_sort(1,n);
if(k<=0||k>n)puts("-1");else
writeln(a[k]);
}
return 0;
}

<pre name="code" class="cpp">//直接调用C++ STL函数 nth_element（start,start+k,end）;  33ms

#include<cstdio>
#include<algorithm>
//#include<bits/stdc++.h>
using namespace std;
template<class T>inline T read(T&x)
{
char c;
while((c=getchar())<=32)if(c==EOF)return 0;
bool ok=false;
if(c=='-')ok=true,c=getchar();
for(x=0; c>32; c=getchar())
x=x*10+c-'0';
if(ok)x=-x;
return 1;
}
template<class T> inline T read_(T&x,T&y)
{
return read(x)&&read(y);
}
template<class T> inline T read__(T&x,T&y,T&z)
{
return read(x)&&read(y)&&read(z);
}
template<class T> inline void write(T x)
{
if(x<0)putchar('-'),x=-x;
if(x<10)putchar(x+'0');
else write(x/10),putchar(x%10+'0');
}
template<class T>inline void writeln(T x)
{
write(x);
putchar('\n');
}
//-------ZCC IO template------
const int maxn=1000011;
const double inf=999999999;
#define lson (rt<<1),L,M
#define rson (rt<<1|1),M+1,R
#define M ((L+R)>>1)
#define For(i,t,n) for(int i=(t);i<(n);i++)
typedef long long  LL;
typedef double DB;
typedef pair<int,int> P;
#define bug printf("---\n");
#define mod 10007
LL a[maxn];
LL k;
int main()
{
LL n,m,i,j,t;
while(read_(n,k))
{
For(i,0,n)
{
read(a[i]);
}
nth_element(a,a+k-1,a+n);
if(k<0||k>n)puts("-1");else
writeln(a[k-1]);
}
return 0;
}