保安站岗[树形DP]

传送门

f[u][0] , [1] , [2] 分别表示选当前 , 选儿子 , 选父亲

f[u] = \sum min(f[son][0],f[son][1],f[son][2])

f[u][1]=\sum min(f[son][0],f[son][1])(at ,least ,1f[son][0])

f[u][2]=\sum min(f[son][0],f[son][1])

我们判断一下选f[son][0]没有

如果没有就加上一个f[son][0]-f[son][1] 最大的 


#include
#define N 2050
#define M N*2
#define inf 0x3fffffff
using namespace std;
int first[N],next[M],to[M],tot;
int n,f[N][3],val[N]; // f[0] itself , f[1] son , f[2] father
// f[u][0] = cigma min(f[son][0],f[son][1],f[son][2]) + val[u]
// f[u][1] = 至少一个f[son][0] + cigme min(f[son][0],f[son][1])
// f[u][2] = cigma min(f[son][0],f[son][1])
void add(int x,int y){
	next[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u,int fa){
	f[u][0] = val[u];
	 int flag=0 , Min=inf;
	for(int i=first[u];i;i=next[i]){
		int t=to[i]; if(t==fa) continue;
		dfs(t,u);
		f[u][0] += min(f[t][0] , min(f[t][1],f[t][2]));
		if(f[t][0] < f[t][1]) flag=1;
		else Min = min(Min , f[t][0]-f[t][1]);
		f[u][1] += min(f[t][0],f[t][1]);
		f[u][2] += min(f[t][0],f[t][1]);
	}
	if(!flag) f[u][1] += Min;
}
int main(){
	scanf("%d",&n); 
	for(int i=1;i<=n;i++){
		int x; scanf("%d",&x); scanf("%d",&val[x]);
		int k; scanf("%d",&k);
		while(k--){int y; scanf("%d",&y); add(x,y),add(y,x);}
	}
	dfs(1,0); printf("%d",min(f[1][0],f[1][1])); return 0;
} 

 

你可能感兴趣的:(树形DP)