ICPC2020 南京 M Monster Hunter

题意：
牛客ICPC2020南京的M题

这道题考场上写了两个小时，奈何最近dp练习太少，写了半天把自己绕晕了，现在来重新捋一下思路。

首先很明确的一点，使用的药水要从小往大进行dp（尝试过从大到小，没办法列方程）。先观察一下要些什么状态：当前节点，以他为根的子树用了多少瓶药，他自己是否用药被杀。最后一个状态我在考场上设立的是是否被杀，其实本质上是一样的。

下面来解释dp的含义：$dp[i][j][0/1]$以$i$为根的子树中，用$j$瓶药，$i$是否被用药杀的状态下（0代表没有），将整个子树消灭的最小代价。

然后可以用经典的模型来解决（我竟然不熟悉了）。

将目前的全集分成两个子集，然后相互组合（如果子集逐渐变大的话，一个点的处理是可以做到O(n)的，否则就是O(n^2）的dp）

对应代码里面的是：先处理Size[u]和Size[v]，再Size[u]+=Size[v]，否则会TLE。

#include<bits/stdc++.h>
#define For(aa, bb, cc) for(int aa = (bb); aa <= (int)(cc); ++aa)
using namespace std;
typedef long long LL;
const int maxn = 2e3 + 10;
const LL inf = 1e15;
int n;
LL dp[maxn][maxn][2];
//"已更新的状态下，以i为根的子树中，用了j瓶药，自己是否用药"的情况下，需要多少代价才能全部清除掉
LL g[maxn][2];
int be[maxn], ne[maxn << 1], to[maxn << 1], e;
int Size[maxn];
LL hp[maxn];

inline void add_edge(int x,int y){
	to[++e] = y, ne[e] = be[x], be[x] = e;
	to[++e] = x, ne[e] = be[y], be[y] = e;
}

void dfs(int u, int f){
	Size[u] = 1;
	if(!be[u]){
		dp[u][0][0] = hp[u];
		dp[u][0][1] = inf;
		dp[u][1][0] = inf;
		dp[u][1][1] = 0;
		return ;
	}
	for(int i = be[u]; i; i = ne[i]){
		int v = to[i];
		if(v == f) continue;
		dfs(v, u);
		For(j, 0, Size[u] + Size[v]) g[j][0] = g[j][1] = inf;
		For(j, 0, Size[u]){
			For(k, 0, Size[v]){
				if(j + k < Size[u] + Size[v]){
					g[j + k][0] = min(g[j + k][0], dp[u][j][0] + dp[v][k][0] + hp[v]);
					if(k > 0) g[j + k][0] = min(g[j + k][0], dp[u][j][0] + dp[v][k][1]);
				}
				if(j > 0){
					g[j + k][1] = min(g[j + k][1], dp[u][j][1] + dp[v][k][0]);
					if(k > 0) g[j + k][1] = min(g[j + k][1], dp[u][j][1] + dp[v][k][1]);
				}
			}
		}
		Size[u] += Size[v];
		For(j, 0, Size[u]) dp[u][j][0] = g[j][0], dp[u][j][1] = g[j][1];
	}
	For(i, 0, Size[u] - 1) dp[u][i][0] += hp[u];
	/*cout<<Size[u]<<endl;
	For(i, 0, Size[u]){
		cout<<u<<" "<<i<<endl;
		cout<<"dp[u][i][0]: "<<dp[u][i][0]<<endl;
		cout<<"dp[u][i][1]: "<<dp[u][i][1]<<endl;
	}
	cout<<"****"<<endl;*/
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	int _, x;
	scanf("%d", &_);
	while(_--){
		scanf("%d", &n);
		e = 0;
		For(i, 1, n) be[i] = Size[i] = 0;
		For(i, 1, n) For(j, 0, n) dp[i][j][0] = dp[i][j][1] = 0;
		For(i, 2, n){
			scanf("%d", &x);
			add_edge(x, i);
		}
		For(i, 1, n) scanf("%lld", &hp[i]);
		dfs(1, 0);
		printf("%lld",dp[1][0][0]);
		For(i, 1, n){
			printf(" %lld", min(dp[1][i][0], dp[1][i][1]));
		}
		puts("");
	}
	return 0;
}