poj 2342 && hdu 1520 树形dp

题意：有n个人，接下来n行是n个人的价值，再接下来n行给出l，k说的是l的上司是k，这里注意l与k是不能同时出现的

链接：点我

dp[i][1] += dp[j][0],

dp[i][0] += max{dp[j][0],dp[j][1]};其中j为i的孩子节点。

第二次做感觉已经很水了

 

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
using namespace std;
#define MOD 1000000007
const int INF=0x3f3f3f3f;
const double eps=1e-5;
typedef long long ll;
#define cl(a) memset(a,0,sizeof(a))
#define ts printf("*****
");
const int MAXN=6010;
int n,m,tt;
int aa[MAXN];
int tot,head[MAXN],dp[MAXN][2],in[MAXN];
struct Edge
{
    int to,next;
}edge[MAXN];
void addedge(int u,int v)
{
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
int dfs(int u)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].to;
        dfs(v);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][0],dp[v][1]);
    }
    return max(dp[u][0],dp[u][1]);
}
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
    cl(dp);
    cl(in);
}
int main()
{
    int i,j,k;
    #ifndef ONLINE_JUDGE
    freopen("1.in","r",stdin);
    #endif
    while(scanf("%d",&n)!=EOF)
    {
        init();
        for(i=1;i<=n;i++)
           scanf("%d",&dp[i][1]);
        int a,b;
        while(scanf("%d%d",&a,&b))
        {
            if(a==0&&b==0)break;
            addedge(b,a);
            in[a]++;
        }
        int ans=0;
        for(i=1;i<=n;i++)        //题目的意思就一棵树
        {
            if(!in[i])  ans=dfs(i);
        }
        printf("%d
",ans);
    }
}