二分——無序數組快速查找第K小的數

#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

AC代碼：快排思想

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<string>

using namespace std;

void swap(int *a, int *b)

{

int t;

t = *a;

*a = *b;

*b = t;

}

int findK(int a[], int k, int s, int e)

{

int m = a[s];

int num = 0, count = s;

for (int i = s + 1; i <= e; i++) {

if (a[i] < m)

{

count++;

num++;

swap(&a[i], &a[count]);

}

}

swap(&a[s], &a[count]);

if (num == k - 1)

return m;

else if (num < k - 1)

return findK(a, k - num - 1, count + 1, e);

else

return findK(a, k, s,count - 1);

}

int main()

{

int N, K;

int a[20009];

while (cin >> N >> K)

{

for (int i = 0; i < N; i++)

cin >> a[i];

if (K >= 1 && K <= N)

printf("%d\n", findK(a, K, 0, N - 1));

else

cout << -1 << endl;

}

return 0;

}