bzoj 3836 [Poi2014]Tourism 状压dp+树形dp

题面

题目传送门

解法

状压dp+树形dp好题

• 图可能形成多个连通块，对于每一个连通块单独处理，因为相互之间不影响答案。
• 那么考虑对于一个连通块的答案应该怎么计算。
• 因为整个图满足最长路不超过10，那么就意味着dfs时候建出的搜索树深度不超过10，并且非树边必然为返祖边。
• 我们不妨假设$0$表示这个点被选择了，$1$表示这个点没有被选择并且它还没有被覆盖，$2$表示这个点没有被选择但是已经被覆盖过。
• 可以考虑设计这样一个状态：$f[i][S]$表示点$i$在搜索树上祖先的选择情况($S$为一个三进制数，当然包括自己)。然后考虑怎么进行转移。
• 先处理祖先对该点的影响，假设$i$的父亲为$j$，到$j$时的集合为$S^{\prime }$。先假设这个点不取，那么只要$S^{\prime }$中和$i$连接的点中有一个为$0$，那么$S$的第$dep[i]$位显然就为$2$，否则为$1$，也就是说 f [ i ] [ S ] = m i n { f [ j ] [ S ′ ] } f[i][S]=min\{f[j][S']\} f[i][S]=min{f[j][S′]}($S$的取值应该根据$S^{\prime }$来定，主要是第$dep[i]$位上的状态)。然后如果这个点被选择了，那么$S^{\prime }$中所有与$i$相连并且为$1$的都可以修改为$2$，即 f [ i ] [ S ] = m i n { f [ j ] [ S ′ ] } + v a l [ i ] f[i][S]=min\{f[j][S']\}+val[i] f[i][S]=min{f[j][S′]}+val[i]。
• 然后再考虑如何处理$i$的子树对它的贡献。假设$j$为$i$的某一个儿子，那么$f[i][S]=min(f[j][S],f[j][S+2\times 3^{dep[i]}])$，就是保证子树合法的情况下尽量最小。
• 现在，时间复杂度为$O((n+m)3^{10})$，应该已经不太好优化了。但是空间复杂度为$O(n\times 3^{10})$，开不下整个$f$数组。
• 考虑如何解决这一问题。可以发现，在$dfs$的过程中，还在栈中的点必然深度两两不同，那么一条路径上总共最多就只有$10$个点会产生贡献，所以我们并不需要明确记录点的编号是什么，只需要记录深度就可以表示这一个状态了。
• 所以现在空间复杂度就变成了$O(10\times 3^{10})$，可以通过。
• 至于为什么不会T，我也不太清楚，可能这种做法在常数上有着优势?或者状态不满？

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
template <typename T> void chkmax(T &x, T y) {x = x > y ? x : y;}
template <typename T> void chkmin(T &x, T y) {x = x < y ? x : y;}
template <typename T> void read(T &x) {
	x = 0; int f = 1; char c = getchar();
	while (!isdigit(c)) {if (c == '-') f = -1; c = getchar();}
	while (isdigit(c)) x = x * 10 + c - '0', c = getchar(); x *= f;
}
const int N = 25010; const ll inf = 1ll << 60;
int cnt, a[N], d[N], pw[N], val[N], vis[N], head[N]; ll f[10][60000];
struct Edge {int next, num;} e[N * 3];
void add(int x, int y) {e[++cnt] = (Edge) {head[x], y}; head[x] = cnt;}
void dfs(int x) {
	vis[x] = 1; int t = d[x];
	if (t) {
		int len = 0;
		for (int p = head[x]; p; p = e[p].next) {
			int k = e[p].num;
			if (vis[k]) a[++len] = d[k];
		}
		for (int s = 0; s < pw[t + 1]; s++) f[t][s] = inf;
		for (int s = 0; s < pw[t]; s++) {
			int fl = 1, cur = s;
			if (f[t - 1][s] == inf) continue;
			for (int i = 1; i <= len; i++) {
				int tmp = (s / pw[a[i]]) % 3;
				if (!tmp) fl = 2; if (tmp == 1) cur += pw[a[i]];
			}
			chkmin(f[t][s + fl * pw[t]], f[t - 1][s]);
			chkmin(f[t][cur], f[t - 1][s] + val[x]);
		}
	} else f[t][0] = val[x], f[t][1] = 0, f[t][2] = inf;
	for (int p = head[x]; p; p = e[p].next) {
		int k = e[p].num; if (vis[k]) continue;
		d[k] = d[x] + 1, dfs(k);
		for (int s = 0; s < pw[t + 1]; s++)
			f[t][s] = min(f[t + 1][s], f[t + 1][s + 2 * pw[t + 1]]);
	}
}
int main() {
	int n, m; read(n), read(m);
	for (int i = 1; i <= n; i++) read(val[i]);
	for (int i = 1; i <= m; i++) {
		int x, y; read(x), read(y);
		add(x, y), add(y, x);
	}
	pw[0] = 1;
	for (int i = 1; i <= 10; i++) pw[i] = 3ll * pw[i - 1];
	memset(vis, 0, sizeof(vis)); ll ans = 0;
	for (int i = 1; i <= n; i++)
		if (!vis[i]) dfs(i), ans += min(f[0][0], f[0][2]);
	cout << ans << "\n";
	return 0;
}