hiho一下 第三十七周二分·二分查找之k小数

来源:互联网 发布:梦幻西游手游mac客户端 编辑:程序博客网 时间:2024/06/04 19:56

描述

在上一回里我们知道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 41732 4176 2602 6176 1303 6207 3125 1 1011 6600
样例输出
1732

问题解析:

注意几个点:

1.a[i]范围大小需用用到long long

2. 思路是采用快速排序的partition思想通过partition快速将数组分为两部分(二分思想)

3.注意写完代码进行一下测试 首先是 N < = 0 或者 K <=0 或者 N<K ,还有就是数组升序或者降序的情况。

如果以上几点都考虑到了就不会有问题了。

/* * Main.cpp * *  Created on: 2015年3月15日 *      Author: jason */#include <iostream>using namespace std;const int MAXLENGTH= 1000000;long long a[MAXLENGTH];int N,k;int Partition(long long a[],int l,int h) {int i = l, j = h;long long x = a[i];    while(i<j) {    while(j>i&&a[j]>x)    j--;    if(i<j)    a[i] = a[j];    while(i<j&&a[i]<x)    i++;    if(i<j)    a[j] = a[i];    }    a[i] = x;    int mid = i;    return mid;}int SeekMinKth(long long a[],int l,int h,int k) {    int mid = Partition(a,l,h);int leftNumbers = mid - l;    if(leftNumbers==k-1) return a[mid];    else if(leftNumbers>k-1)    return SeekMinKth(a,l,mid-1,k);    elsereturn SeekMinKth(a,mid+1,h,k-leftNumbers-1);}int main(){    while(cin>>N>>k) {    for(int i=0;i<N;i++)    cin>>a[i];if(N<=0||N<k||k<=0)cout<<"-1";    else cout<<SeekMinKth(a,0,N-1,k);    }return 0;}


0 0
原创粉丝点击