HDU 1520

跟HDU 1054差不多。水

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=6005,inf=1<<30;
int dp[maxn][2];
int w[maxn],in[maxn];
int n,m;
vector<int>G[maxn];
void dfs(int u,int fa)
{

    dp[u][1]=w[u];
    dp[u][0]=0;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][1],dp[v][0]);
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        int rt=1;
        for(int i=0;i<=n;i++) G[i].clear(),in[i]=0;
        for(int i=1;i<=n;i++) scanf("%d",&w[i]);
        int from,to;
        while(scanf("%d%d",&from,&to),from+to)
        {
            G[to].push_back(from);//建一条上司到下属的边
            in[from]++;
        }
        for(int i=1;i<=n;i++)
        if(in[i]==0) {rt=i;break;}
        dfs(rt,-1);
        int ans=max(dp[rt][0],dp[rt][1]);
        printf("%d\n",ans);
    }
    return 0;
}