CF1463-D. Pairs

CF1463-D. Pairs

题意：

有从\(1\)到\(2n\)一共\(2n\)个数字，让你将这\(2n\)个数字分成\(n\)组，每组有两个数字。对于这\(n\)组数字，你可以从中挑选\(x\)组做\(min\)操作，其他的\(n-x\)组中做\(max\)操作，这样就可以得到一个新的数组\(b\)； 现在题目给你得到的数组\(b\)，问你可以有多少不同的\(x\)使得可以得到数组\(b\)。

---

思路：

我们从这\(2n\)个数字中去掉数组\(b\)中的数，剩下的就是数组\(a\)中的数。对\(a\)，\(b\)数组排序之后，我们现在先枚举每个\(x\)让\(a\)中最大的\(x\)个数字从小到达与\(b\)中最小的x个数字从小到大组合，让\(a\)中剩余的\(n-x\)个数字从小到大与\(b\)中剩余的\(n-x\)个数字从小到大组合（类似于贪心的思想），这个应该是最优的情况，如果这样还是不能通过前\(x\)个取\(min\)后\(n-x\)取\(max\)得到数组\(b\)，那么对于这个\(x\)无论你再怎么组合都不可能得到数组\(b\)。

从理论上来说上面这种枚举+贪心的方法肯定能得到最终的答案，但是时间复杂度达到了\(o(n^2)\)，这是不能接受的。我们再仔细分析一下，会发现符合要求的\(x\)是连续的、在一个区间里面的，原因如下：

我们假设符合要求的\(x\)的区间为\([L, R]\)。现在我们将\(x=R\)情况对应的组合进行操作可以得到\(x=R+1\)的情况：将数组\(a\)中\(n-x\)个最小的数字中最大的一个数字（称它为\(i\)）与数组\(b\)中\(x\)个最小的数字中最小的一个数字（称它为\(j\)）进行组合，这时候一定是因为\(i<j\)从而取\(min\)操作时不能得到\(j\)所以不符合条件。而对于之后的\(x=R+1, ..., x=n\)的情况，\(b\)中\(x\)个数字最小的数字中最小的数字是不变的，而\(a\)中\(x\)个最小的数字是不断变小的，所以之后的情况也都是不符合的。同理我们也可以从\(x=L-1,..., x=0\)这些情况中得到同样的结论。

通过这个结论，我们可以通过两次二分查找，找到符合条件的\(x\)区间\([L, R]\)的\(L\)和\(R\)，这样就可以将时间复杂度优化到\(o(nlogn)\)。

AC代码

#include <cstdio>
#include <cstring>
#include <algorithm>

const int maxn = 2e5 + 5;

int a[maxn], b[maxn];

int check(int mid, int n) { // 0 suit;  1 l = mid + 1; 2 r = mid - 1;
	for (int i = 0; i < mid; i++) {
		if (b[i] > a[n - mid + i]) {
			return 2;
		}
	}
	for (int i = 0; i < n - mid; i++) {
		if (a[i] > b[mid + i]) {
			return 1;
		}
	}
	return 0;
}

int main() {
	int T, n;
	scanf("%d", &T);
	while (T--) {
		scanf("%d", &n);
		for (int i = 0; i < n; i++) {
			scanf("%d", &b[i]);
		}
		std::sort(b, b + n);
		int tot = 0, cur = 0;
		for (int i = 0; i < 2 * n; i++) {
			if (i + 1 == b[cur]) {
				cur++;
			} else {
				a[tot++] = i + 1;
			}
		}
		int l = 0, r = n;
		while (l <= r) {
			int mid = (l + r) >> 1;
			if (check(mid, n) != 1) {
				r = mid - 1;
			} else {
				l = mid + 1;
			}
		}
		int L = l;
		l = 0, r = n;
		while (l <= r) {
			int mid = (l + r) >> 1;
			if (check(mid, n) != 2) {
				l = mid + 1;
			} else {
				r = mid - 1;
			}
		}
		int R = r;
		printf("%d\n", R - L + 1);
	}

	return 0;
}