给定棵树和每个节点的权值,相邻父子节点的不可重复选取,求其可以取到的最大权值
#include<bits/stdc++.h>
using namespace std;
const int MAX=6e3+5;
int n,a[MAX],d[MAX];
struct P
{
int to,nxt;
}e[MAX];
int head[MAX],tot;
long long dp[MAX][2];
void init()
{
memset(d,0,sizeof(d));
memset(dp,0,sizeof(dp));
memset(head,-1,sizeof(head));
tot=0;
}
void adde(int u,int v)
{
e[tot].to=v;
e[tot].nxt=head[u];
head[u]=tot++;
}
void dfs(int u)
{
dp[u][0]=0;
dp[u][1]=a[u];
for(int i=head[u];~i;i=e[i].nxt)
{
int v=e[i].to;
dfs(v);
dp[u][0]+=max(dp[v][1],dp[v][0]);
dp[u][1]+=dp[v][0];
}
}
int main()
{
int x,y;
while(~scanf("%d",&n))
{
init();
for(int i=1;i<=n;++i) scanf("%d",&a[i]);
while(~scanf("%d%d",&x,&y))
{
if(x==0&&y==0) break;
adde(y,x);++d[x];
}
int rt=0;
for(int i=1;i<=n;++i) if(!d[i]) {dfs(rt=i);break;}
printf("%lld\n",max(dp[rt][0],dp[rt][1]));
}
return 0;
}