【树形DP】hdu 1520 Anniversary party

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int NM=6005;
int dp[NM][2],f[NM],a[NM];
vector<int>v[NM];

int max(int x,int y){
return x>y?x:y;
}

void DFS(int s){
int i;
dp[s][1]=a[s];
for(i=0;i<v[s].size();i++){
DFS(v[s][i]);
}
for(i=0;i<v[s].size();i++){  //
dp[s][0]+=max(dp[v[s][i]][0],dp[v[s][i]][1]);
dp[s][1]+=dp[v[s][i]][0];
}
}

int main()
{
int i,n,u,L,K;
while(~scanf("%d",&n)){
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
v[i].clear();
}
memset(f,0,sizeof(f));
memset(dp,0,sizeof(dp));
while(scanf("%d%d",&L,&K)){
if(L==0 && K==0) break;
f[L]=K;
v[K].push_back(L);
}
for(i=1;i<=n;i++){
if(!f[i]){
u=i;break;
}
}
DFS(u);
printf("%d\n",max(dp[u][0],dp[u][1]));
}
return 0;
}