HDU 1520 Anniversary party树状DP

最简单的树状DP了。。

这题的特殊性可以建成无向图，以任意节点为根节点。。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 6010
int val[maxn];
int dp[maxn][2];
bool vis[maxn];
vector<int> vec[maxn];
void init()
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<maxn;i++)
    vec[i].clear();
}
void dfs(int rt)
{
    vis[rt]=1;
    int sz=vec[rt].size();
    if(sz==1&&rt!=1) return;
    for(int i=0;i<sz;i++)
    {
        int tmp=vec[rt][i];
        if(!vis[tmp])
        {
            dfs(tmp);
            dp[rt][1]+=dp[tmp][0];
            dp[rt][0]+=max(dp[tmp][0],dp[tmp][1]);
        }
    }
}

int main()
{
    freopen("in.txt","r",stdin);
    int n;
    while(scanf("%d",&n)==1)
    {
        init();
        int i;
        for(i=1;i<=n;i++)
        scanf("%d",&val[i]);
        for(i=1;i<=n;i++)
        {
            dp[i][0]=0;
            dp[i][1]=val[i];
        }
        int u,v;
        while(scanf("%d%d",&u,&v)==2&&(u||v))
        {
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        dfs(1);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}