【慢慢学算法】：查找第k小数

题目描述：

查找一个数组的第K小的数，注意同样大小算一样大。 
如  2 1 3 4 5 2 第三小数为3。

输入：

输入有多组数据。
每组输入n，然后输入n个整数(1<=n<=1000)，再输入k。

输出：

输出第k小的整数。

样例输入：

6
2 1 3 5 2 2
3

样例输出：

3

 

#include<iostream>
#include<fstream>
#include<algorithm>
using namespace std;

int main()
{
    int n;
    while(cin >> n)
    {
	int a[n];
	int flag = 1;   //注意flag
	int k;
	for(int i = 0; i < n; i++)
	    cin >> a[i]; 
	cin >> k;
	sort(a, a+n);
	for(int i = 1; i < n; i++)   //该循环的处理
	{
	    if(a[i-1] != a[i])
		flag++;
	    if(flag == k)
	    {
		cout << a[i] << endl;
		break;
	    }
	}
	
    }
}