欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

洛谷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;
}