Content

给定一个 \(n\) 个数的数列 \(a_1,a_2,a_3,...,a_n\) 和一个数 \(k\)，试找出这样的一个数 \(x\)，使得数列中有 \(k\) 个数小于等于 \(x\)。求出任意一个 \(x\)，或者 \(x\) 不存在。

数据范围：\(1\leqslant n\leqslant 2\times 10^5,0\leqslant k\leqslant n,1\leqslant a_i\leqslant 10^9,x\in[1,10^9]\)。

Solution

比大小的话肯定要排序，所以我们先对这个序列排序。设排序后的每个数下标是按照其排序之后的位置决定的。然后，按照 \(k\) 的值分类讨论：

• \(k=0\)，那么看最小的数 \(a_1\) 是否大于 \(1\)，是的话 \(x=a_1-1\)，否则不存在这样的 \(x\)。
• \(k>0\)，那么直接看是否有 \(a_k=a_{k+1}\)，有的话 \(x\) 不存在，否则 \(x=a_k\)。

Code

int n, k, a[200007], ans;

int main() {
	getint(n), getint(k);
	_for(i, 1, n)	getint(a[i]);
	sort(a + 1, a + n + 1);
	ans = (!k ? (a[1] > 1 ? a[1] - 1 : -1) : (a[k] == a[k + 1] ? -1 : a[k]));
	writeint(ans);
	return 0;
}