hdu 4340 树形dp

/*

题目：http://acm.hdu.edu.cn/showproblem.php?pid=4340

题意：两个人进攻n个城市的国家，这n个城市构成一棵树，可以任意选择一个点开始，攻击已被自己攻击过的点的相邻点，时间可以减半，两个人的攻击时间不一

这个状态要求这个连通块有一个入口嘛

然后这个入口可以是i这个节点

也可以使他的子节点，或间接的子节点

如果选择i作为入口就+a[i]

否则就在子节点中挑已经有入口的状态，然后加上a[i]/2

当然选择较优的那种

为此状态转移方程为：

dp[u][0]=min(dp[v][0],dp'[v][1])+a[i]/2;

dp[u][1]=min(dp[v][0],dp'[v][1])+min(a[i],a[i]/2+min(dp[v][1]-min(dp[v][0],dp'[v][1])));

*/

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<memory.h>
using namespace std;
const int maxn=202;
const int inf=0x3fffff;
int n,NE,dpa[maxn][2],dpb[maxn][2],a[maxn],b[maxn],head[maxn];
struct node
{
int u,v,next;
}Edge[maxn];
void addEdge(int u,int v)
{
Edge[NE].u=u;
Edge[NE].v=v;
Edge[NE].next=head[u];
head[u]=NE++;
}
void dfs(int u,int fa)
{
int i,j,v,tmp,leaf=1,ma=inf,mb=inf;
for(i=head[u];i!=-1;i=Edge[i].next)
{
v=Edge[i].v;
if(v==u) continue;
leaf=0;
dfs(v,u);
tmp=min(dpa[v][0],dpb[v][1]);
ma=min(ma,dpa[v][1]-tmp);
dpa[u][0]+=tmp;
tmp=min(dpa[v][1],dpb[v][0]);
mb=min(mb,dpb[v][1]-tmp);
dpb[u][0]+=tmp;
}
if(leaf)
{
dpa[u][0]=a[u]/2;dpa[u][1]=a[u];
dpb[u][0]=b[u]/2;dpb[u][1]=b[u];
}
else
{
dpa[u][1]=dpa[u][0]+min(a[u],a[u]/2+ma);
dpb[u][1]=dpb[u][0]+min(b[u],b[u]/2+mb);
dpa[u][0]+=a[u]/2;
dpb[u][0]+=b[u]/2;
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
NE=0;
memset(dpa,0,sizeof(dpa));
memset(dpb,0,sizeof(dpb));
memset(head,-1,sizeof(head));
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
addEdge(a,b);
}
dfs(1,0);
printf("%d\n",min(dpa[1][1],dpb[1][1]));
}
return 0;
}