#树形dp，二次扫描换根法#poj 3585 Accumulation Degree

题目

求一棵树的最大流

---

分析

这道题可以用树形dp解决，以任意一个点为源点求出每个点的流量，再用二次扫描换根法找到最优的根
流量求法：

if (deg[e[i].y]==1) d[x]+=e[i].w; else d[x]+=min(d[e[i].y],e[i].w);//度为1就是边权，否则要综合子节点去考虑

动态规划：

if (deg[x]==1) f[e[i].y]=d[e[i].y]+e[i].w;//度只有1
		else f[e[i].y]=d[e[i].y]+min(f[x]-min(d[e[i].y],e[i].w),e[i].w);//度有多个需要判断最小值，并且换根的差值也要算进去

---

代码

#include <cstdio>
#include <cstring>
struct node{int y,w,next;}e[400001]; bool v[200001]; 
int d[200001],f[200001],ls[200001],deg[200001],t,n;
int in(){
	int ans=0; char c=getchar();
	while (c<48||c>57) c=getchar();
	while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
	return ans;
}
int max(int a,int b){return (a>b)?a:b;}
int min(int a,int b){return (a<b)?a:b;}
void dp(int x){//树形dp
	d[x]=0; v[x]=1;
	for (register int i=ls[x];i;i=e[i].next)
	if (!v[e[i].y]){
		dp(e[i].y);
		if (deg[e[i].y]==1) d[x]+=e[i].w; else d[x]+=min(d[e[i].y],e[i].w);
	}
}
void dfs(int x){//二次扫描换根法
	v[x]=1;
	for (register int i=ls[x];i;i=e[i].next)
	if (!v[e[i].y]){
		if (deg[x]==1) f[e[i].y]=d[e[i].y]+e[i].w;
		else f[e[i].y]=d[e[i].y]+min(f[x]-min(d[e[i].y],e[i].w),e[i].w);
		dfs(e[i].y);
	}
}
int main(){
	t=in();
	while (t--){
		n=in(); memset(ls,0,sizeof(ls)); memset(deg,0,sizeof(deg));
		for (register int i=1;i<n;i++){//建树（双向边）
			int x=in(),y=in(); int w=in();
			e[i]=(node){y,w,ls[x]}; ls[x]=i; deg[y]++; deg[x]++;//入度+出度
			e[i+n-1]=(node){x,w,ls[y]}; ls[y]=i+n-1;
		}
		memset(v,0,sizeof(v)); int root=1;
		dp(root); memset(v,0,sizeof(v));
		f[root]=d[root]; dfs(root);//一定要注意一开始f[root]=d[root]
		int ans=0;
		for (register int i=1;i<=n;i++) ans=max(ans,f[i]);
		printf("%d\n",ans);
	}
	return 0;
}