POJ_2342_树状dp

http://poj.org/problem?id=2342

第一道树状dp，入门题，用vector构建有向图。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
int happy[],last[],dp[][];
vector<int> v[];
vector<int>::iterator it;
void dfs(int now)
{
    dp[now][] = happy[now];
    int len = v[now].size();
    for(int i = ;i < len;i++)  dfs(v[now][i]);
    for(int i = ;i < len;i++)
    {
        dp[now][] += dp[v[now][i]][];
        dp[now][] += max(dp[v[now][i]][],dp[v[now][i]][]);
    }

}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        memset(dp,,sizeof(dp));
        memset(last,-,sizeof(last));
        for(int i = ;i <= n;i++)   scanf("%d",&happy[i]);
        int a,b;
        while(scanf("%d%d",&a,&b) && a && b)
        {
            v[b].push_back(a);
            last[a] = b;
        }
        a = ;
        while(last[a] != -)    a = last[a];
        dfs(a);
        printf("%d\n",max(dp[a][],dp[a][]));
    }
    return ;
}