2021-2022-2 ACM集训队每周程序设计竞赛（12） - 问题 F: 树上路径染色 - 题解

题意：

一颗 $n$ 个节点、$n-1$ 条无向边的树，可以将树上的边任意染成黑色或者是白色。
给出 $m$ 对点 $u、v$，对于每一对点，都要求从 $u$ 到 $v$ 的通路上，至少有一条黑边。
求符合 $m$ 对点的树的个数。

思路：

首先，所有的染色方案为 $a=2^{n-1}$，但是所有合法的方案很难求，所以我们换个角度，求所有不合法的方案 $b$，最终答案就是 $a-b$。

假设不符合第 $i$ 对点的情况为 $S_i$ ，那么 $b=\bigcup _{i=1}^m{S}_i$。这时候思路是不是突然就很敞亮敞亮，该用容斥了。设$S_i$ 中的方案数量为 $\left|S_i\right|$，则$$b=\bigcup _{i=1}^m{S}_i=\sum _{i=1}^m\left|S_i\right|-\sum _{i,j;1\le i<j\le m}\left|S_i\cap S_j\right|+\dots \dots +(-1{)}^{m-1}\ast \left|S_i\cap \dots \dots \cap S_m\right|$$
现在只剩下一个问题没有解决了，那就是不合法方案 $S_i$ 的计算。
这时候注意一下 $n$ 的范围，我们完全可以用二进制状态压缩的方法去存从根节点到任意节点的通路所经过的边 $d_u$，那么 $d_u\oplus d_v$ 就表示从节点 $u$ 到节点 $v$ 的通路所经过的边。这些边必然要染成白色，那么那些多余的边，就可以任意染色。如果 $d_i$ 中有 $cnt$ 条边的话， $\left|S_i\right|=2^{n-1-cnt}$ 。
单个数对的不合法条件数可以求出，那么对于多个不合法条件的并集，只需要对通路中的路径求并后，之后的处理类似。

时间复杂度：$O（2^m）$

#include <bits/stdc++.h>
using namespace std;
int n, m;
long long s[55], d[55];
int h[1010], e[1010], ne[1010], idx;

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs(int u, int fa, long long cnt) {
	s[u] = cnt;

	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		if (j == fa)
			continue;

		dfs(j, u, cnt | (1ll << j));
	}
}

int main() {

	cin >> n;
	for (int i = 1; i <= n; i++)
		h[i] = -1;

	for (int i = 1, a, b; i < n; i++) {
		cin >> a >> b;
		add(a, b), add(b, a);
	}
	dfs(1, -1, 0ll);

	cin >> m;
	for (int i = 0, a, b; i < m; i++) {
		cin >> a >> b;
		d[i] = s[a] ^ s[b];
	}

	long long res = 1ll << (n - 1);
	for (int i = 1; i < (1ll << m); i++) {
		long long num = 0ll;
		int cnt = 0;
		for (int j = 0; j < m; j++)
			if (i & (1 << j)) {
				cnt++;
				num |= d[j];
			}

		int tmp = n - 1;
		while (num) {
			if (num & 1)
				tmp--;
			num >>= 1;
		}

		if (cnt & 1)
			res -= (1ll << tmp);
		else
			res += (1ll << tmp);
	}

	cout << res;
	return 0;
}