P1138 第k小整数 题解

题目描述

现有 $n$ 个正整数，要求出这 $n$ 个正整数中的第 $k$ 个最小整数（相同大小的整数只计算一次）。

输入格式

第一行为 $n$ 和 $k$；第二行开始为 $n$ 个正整数的值，整数间用空格隔开。

输出格式

第 $k$ 个最小整数的值；若无解，则输出 NO RESULT。

样例

样例输入

10 3
1 3 3 7 2 5 1 2 4 6

样例输出

3

数据范围与提示

$n\le 10000$，$k\le 1000$，正整数均小于 $30000$。

思路及部分实现

笔者的思路是读入数据后用头文件库中的 $sort$ 函数将数组从小到大排序，然后依次枚举每个数：

• 如果当前的数字与上一个数字相同，说明是重复数字，因此不做考虑
• 如果当前的数字与上一个数字不相同，说明不是重复数字，将计数器加一
• 当计数器等于 $k$ 时就输出当前数字，如果整个过程中都没输出说明没有解，输出NO RESULT

具体代码实现如下：

sort(a+1,a+n+1);
for(int i=2;i<=n;i++){
	if(cnt==k){
		cout<<a[i-1];
		//因为是在开头判断计数器，计数器一定是在上一步时就等于 k了，所以输出的是上一个数
       return 0;
    }
    if(a[i]!=a[i-1]) cnt++;
}
printf("NO RESULT");

完整代码

#include<iostrean>
using namespace std;
int a[10005];
int main()
{
    int n,k,cnt=1;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    for(int i=2;i<=n;i++){
        if(cnt==k){
            cout<<a[i-1];
            return 0;
        }
        if(a[i]!=a[i-1]) cnt++;
    }
    printf("NO RESULT");
    return 0;
}