hdu6662 Acesrc and Travel

http://acm.hdu.edu.cn/showproblem.php?pid=6662

之前那几题都没写换根树形DP，好久没写换根了写了好久，初始化还写错了，对着多校数据调了好久。。。

换根树形DP的关键就是dfs2次，先记录子树信息，再dfs一遍更新从父节点过来的信息。

这题先手拿，后手必然会拿子节点中对自己最优的，所以转移是固定的。

那么先dp[u][0]表示先手拿u的点能得到的最大的a-b差值，dp[u][1]表示后手拿u点能得到的最小a-b差值。

然后先一遍DFS算出dp[u][0],dp[u][1]，第二遍dfs再从父节点向下更新。

#include<bits/stdc++.h>
#define maxl 100010
using namespace std;

int n,cnt;
int a[maxl],b[maxl],ehead[maxl];
long long f[maxl][2],dp[maxl][2];
long long fir[maxl][2],sec[maxl][2];
int firid[maxl][2],secid[maxl][2];
struct ed
{
	int to,nxt;
}e[maxl<<1];

inline void add(int u,int v)
{
	e[++cnt].to=v;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void updmi(int u,int v,long long a)
{
	if(firid[u][0]==0 || a<fir[u][0])
	{
		sec[u][0]=fir[u][0];secid[u][0]=firid[u][0];
		fir[u][0]=a;firid[u][0]=v;
	}else if(a<sec[u][0] || secid[u][0]==0)
		sec[u][0]=a,secid[u][0]=v;
}

inline void updmx(int u,int v,long long a)
{
	if(firid[u][1]==0 || a>fir[u][1])
	{
		sec[u][1]=fir[u][1];secid[u][1]=firid[u][1];
		fir[u][1]=a;firid[u][1]=v;
	}else if(a>sec[u][1] || secid[u][1]==0)
		sec[u][1]=a,secid[u][1]=v;
}

inline void dfs1(int u,int fa)
{
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(v==fa)
			continue;
		dfs1(v,u);
		updmi(u,v,f[v][1]);updmx(u,v,f[v][0]);
	}
	f[u][0]=fir[u][0]+a[u]-b[u];
	f[u][1]=fir[u][1]+a[u]-b[u];
	
}

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		ehead[i]=0;
		f[i][0]=f[i][1]=0;dp[i][0]=dp[i][1]=0;
		fir[i][0]=fir[i][1]=firid[i][0]=firid[i][1]=0;
		sec[i][0]=sec[i][1]=secid[i][0]=secid[i][1]=0;
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]);
	cnt=0;int u,v;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	dfs1(1,0);
}

inline void dfs2(int u,int fa)
{
	long long tmx,tmi;
	if(fa!=0)
	{
		tmi=dp[fa][1];
		if(firid[fa][1]==u)
		{
			tmi-=fir[fa][1];
			if(secid[fa][1]>0)
				tmi+=sec[fa][1];
		}
		tmx=dp[fa][0];
		if(firid[fa][0]==u)
		{
			tmx-=fir[fa][0];
			if(secid[fa][0]>0)
				tmx+=sec[fa][0];
		}
		updmi(u,fa,tmi);updmx(u,fa,tmx);
	}
	dp[u][0]=fir[u][0]+a[u]-b[u];
	dp[u][1]=fir[u][1]+a[u]-b[u];
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(v==fa)
			continue;
		dfs2(v,u);
	}

}

inline void mainwork()
{
	dp[1][0]=f[1][0];dp[1][1]=f[1][1];
	dfs2(1,0);
}

inline void print()
{
	long long mx=dp[1][0];
	for(int i=2;i<=n;i++)
		mx=max(mx,dp[i][0]);
	printf("%lld\n",mx);
}

int main()
{
	//freopen("f.in","r",stdin); 
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}
/*
5
2 0 2 2 0
1 1 4 2 0
1 5
2 5
4 1
3 4
5
3 2 0 0 4
4 4 2 1 2
1 2
1 3
4 3
1 5
5
4 0 1 0 3
3 1 4 2 4
5 3
1 3
2 3
5 4
*/