CF241B Friends

link

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

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

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

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

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$ 大的节点，那么预处理 $t{r}_{i,j}$ 表示以 $i$ 为根的子树内存的节点在第 $j$ 位上为 $1$ 的个数，那么有了 $t{r}_{i,j}$ 就方便计算答案了。

• 若 $x$ 第 $i$ 位为 $0$，那么 $k\oplus 1$ 内的所有节点都可以比 $x$ 大，看 $a_i$ 第 $j$ 位上是否为 $1$，若是则为 $cn{t}_0\times 2^j$，否则为 $cn{t}_1\times 2^j$。

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

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