洛谷P1138 第k小整数(STL中的unique去重函数)
程序员文章站
2022-05-14 20:17:08
...
【题目描述】
现有n个正整数,n≤10000,要求出这n个正整数中的第k个最小整数(相同大小的整数只计算一次),k≤1000,正整数均小于30000。
【输入格式】
第一行为n和k
第二行开始为n个正整数的值,整数间用空格隔开。
【输出格式】
第k个最小整数的值;若无解,则输出“NO RESULT”。
【输入样例】
10 3
1 3 3 7 2 5 1 2 4 6
【输出样例】
3
【解题思路】
unique函数将相邻元素间的重复元素全部移动至末端,返回去重之后数组最后一个元素之后的地址
因此unique(a, a + n) - a即为去重数组后的长度
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 10005;
int a[N], n, k;
int main()
{
cin >> n >> k;
for (int i = 0; i < n; i++) cin >> a[i];
sort(a, a + n);
int len = unique(a, a + n) - a;
if (len < k) cout << "NO RESULT" << endl;
else cout << a[k - 1] << endl;
return 0;
}