CF241B Friends

link

题意：求 \(n\) 个数中异或值前 \(m\) 大的数对之和。

考虑先求出第 \(m\) 大的异或值为什么。先建 trie 树，然后二分，直接枚举每个数，在 trie 上贪心的走。

• 若 \(x\) 第 \(i\) 位为 \(0\)，那么 \(k\oplus1\) 内的所有节点都可以比 \(x\) 大，走到 \(k\) 上。

• 若 \(x\) 第 \(i\) 位为 \(1\)，那么考虑走到 \(k\oplus1\) 上，答案尚未不能计算。

int calc(int x) {
	int tot = 0;
	for (int i = 1; i <= n; ++i) {
		int u = 0;
		for (int j = 30; ~j; --j) {
			int t1 = a[i]>>j&1, t2 = x>>j&1;
			if (t2) u = ch[u][t1^1];
			else tot += val[ch[u][t1^1]], u = ch[u][t1];
			if (!u) break;
		}
		tot += val[u];
	}
	return tot/2;
}

求出来 \(kth\)，再考虑求本题答案。若再跑一次过程，那么答案就会出现在可能比 \(x\) 大的节点，那么预处理 \(tr_{i,j}\) 表示以 \(i\) 为根的子树内存的节点在第 \(j\) 位上为 \(1\) 的个数，那么有了 \(tr_{i,j}\) 就方便计算答案了。

• 若 \(x\) 第 \(i\) 位为 \(0\)，那么 \(k\oplus1\) 内的所有节点都可以比 \(x\) 大，看 \(a_i\) 第 \(j\) 位上是否为 \(1\)，若是则为 \(cnt_0\times2^j\)，否则为 \(cnt_1\times2^j\)。

• 若 \(x\) 第 \(i\) 位为 \(1\)，那么考虑走到 \(k\oplus1\) 上，答案尚未不能计算。

void solve(int k) {
	for (int i = 1; i <= n; ++i) {
		int u = 0;
		for (int j = 30; ~j; --j) {
			int t1 = a[i]>>j&1, t2 = k>>j&1;
			if (t2) u = ch[u][t1^1];
			else {
				int t = ch[u][t1^1];
				for (int o = 0; o <= 30; ++o) {
					int t3 = a[i]>>o&1;
					if (t3) ans = (ans+1ll*(val[t]-tr[t][o])*(1ll<<o))%mod;
					else ans = (ans+1ll*tr[t][o]*(1ll<<o))%mod;
				}
				u = ch[u][t1];
			}
			if (!u) break;
		}
		ans = (ans+1ll*val[u]*k)%mod;
	}
}

计算出来的答案需要除 \(2\)，且数出来的答案可能出现若干异或值均为 \(kth\) 的点对，那么就减去多余的部分。

code