分治之第k小的数

Description

输入n个数，求其中第k小的数。

Input

第一行包含两个整数n和k；n<1000，1<=K<=n

第二行包含n个整数。

Output

输出第k小的那个整数。

Sample Input

15 1
1 3 7 2 4 6 -1 0 9 88 2 5 17 6 1

Sample Output

-1

#include<iostream>
using namespace std;

int a[1010];
int n, k;


void swap(int &a, int &b)
{
	int temp = a;
	a = b;
	b = temp; 
}


int partition(int l, int r)
{
	int i = l, j = r+1;
	while(true){
		while(a[++i]<a[l] && i<r);
		while(a[--j]>a[l]);
		if(i>=j)
			break;
		swap(a[i], a[j]);
	}
	swap(a[l], a[j]);
	return j;	
} 


int quick_sort(int l, int r)
{
	if(l<=r){
		int p = partition(l, r);
		if(p == k-1)
			return a[p];
		else if(p>k-1)
			quick_sort(l, p-1);
		else 
			quick_sort(p+1, r);
	}
}


int main()
{
	cin >> n >> k;
	int i = 0;
	while(i<n){
		int t;
		cin >> t;
		a[i++] = t;
	}
	int r = quick_sort(0, n-1);
	cout << r << endl;
	return 0;
}