最快乐的舞会 树形dp
代码:
#include<cstdio>
#include<cstring>
#include <vector>
using namespace std;
int dp[6010][2];
int father[6010];
int visit[6010];
int n;
vector <int> zi[6010];
void dfs(int root)
{
visit[root]=1;
for(int i=0;i<zi[root].size();i++){
if(!visit[zi[root][i]]){
dfs(zi[root][i]);
dp[root][1]+=dp[zi[root][i]][0];
if(dp[zi[root][i]][1]>dp[zi[root][i]][0])
dp[root][0]+=dp[zi[root][i]][1];
else
dp[root][0]+=dp[zi[root][i]][0];
}
}
}
int main()
{
scanf("%d",&n);
memset(dp,0,sizeof(dp));
memset(visit,0,sizeof(visit));
for(int i=1;i<=n;i++){
scanf("%d",&dp[i][1]);
zi[i].clear();
}
for(int i=1;i<=n;i++)
father[i]=i;
int l,k;
while(1){
scanf("%d%d",&l,&k);
if(l==0&&k==0)
break;
father[l]=k;
zi[k].push_back(l);
}
int root=n;
while(root!=father[root]){
root=father[root];
}
dfs(root);
if(dp[root][0]>dp[root][1])
printf("%d\n",dp[root][0]);
else
printf("%d\n",dp[root][1]);
return 0;
}