树形dp poj 2342

上午看了想了好长时间，下午用了一个多小时才写出来，主要对回溯没怎么写过。

第一题 树形dp。

题意就是有关系的两个人不能同时出现。

dp1【s】 表示 s点去。

dp2【s】 表示s点不去。

dp1【s】+=dp2【x】。 //s点去，ze x点不去    （s为父节点）。

dp2【s】+=max（dp1【x】，dp2【x】）；// s点不去，则x去或不去的最大的一个。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
vector<int>V[6006];
int vis[6006];
int dp1[6006];//去
int dp2[6006];//不去
void init()
{
    for(int i=0;i<6005;i++)
      V[i].clear();
    memset(dp1,0,sizeof(dp1));
    memset(dp2,0,sizeof(dp2));
    memset(vis,0,sizeof(vis));
}
void dfs(int s)
{
    vis[s]=1;
    for(int i=0;i<(int)V[s].size();i++)
    {
        int x=V[s][i];
        if(!vis[x])
        {
            dfs(x);
            dp1[s]+=dp2[x];   //s去，则x不去。
            dp2[s]+=max(dp1[x] , dp2[x]);   //s不去，则在s点加上x去或不去的最大值.
            //printf("dp1[%d]=%d   dp2[%d]=%d\n",s,dp1[s],s,dp2[s]);
        }
    }
}
int main()
{
    //freopen("Input.txt","r",stdin);
    int n,i;
    while(~scanf("%d",&n))
    {
        init();
        for(i=1;i<=n;i++)
        {
            scanf("%d",&dp1[i]);
            dp2[i]=0;
        }
        int a,b;
        while(scanf("%d%d",&a,&b)!=EOF)
        {
            if(a==0&&b==0) break;
            V[a].push_back(b);
            V[b].push_back(a);
        }
        for(i=1;i<n;i++)
        {
            if(vis[i]==0)
            dfs(i);
        }
        int res=max(dp1[1],dp2[1]);
        printf("%d\n",res);
    }
    return 0;
}