2342-Anniversary party-树dp

Anniversary party

题目链接： http://poj.org/problem?id=2342

题意：给定每个人的快乐值，邀请一部分人去参加宴会，被邀请的人和该人的上司只能去一个，使得最终的快乐值最大。

应该是一道很简单的树dp吧，正在学习当中 ~ing，树dp，紫薯上的最大独立集啊。

包括两部分，建树部分和dp部分，，，建树、、、、、和dp（找到转移方程啊）

用1，0分别表示去和不去的状态，dp[i][0],dp[i][1],到第i位上司为止的最大的快乐值（这个人去或不去）。

dp[u][1]+=dp[v][0];    //上司去，下属不去
dp[u][0]+=max(dp[v][0],dp[v][1]);//上司不去，下属去或不去的最大值

用dfs依次递归，如果dfs(1,-1)则是从1开始，最终返回到最大值在1,处，如果dfs(n,-1)开始，最终返回到n处最大值，理解递归。

参考代码：

#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <queue>
#include <stack>
#include <iostream>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=6010;
struct Edge
{
    int v,next;
}edge[maxn*2];
int n,k,cnt[maxn],head[maxn],dp[maxn][2];
void add_edge(int u,int v)
{
    edge[k].v=v;
    edge[k].next=head[u];
    head[u]=k++;
}
void dfs(int u,int fa)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v!=fa)
        {
            dfs(v,u);
            dp[u][1]+=dp[v][0];    //去
            dp[u][0]+=max(dp[v][0],dp[v][1]);//不去
        }
    }
}
int  main()
{
    //freopen("C:\\Users\\Administrator\\Desktop\\a.txt","r",stdin);
    k=0;
    int a,b;
    for(int i=0;i<maxn;i++)
        head[i]=-1,dp[i][0]=0,dp[i][1]=0;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&dp[i][1]);
        while(scanf("%d%d",&a,&b),a&&b)
            add_edge(a,b),add_edge(b,a);
        dfs(1,-1);
        cout<<max(dp[1][0],dp[1][1])<<endl;
    }
    return 0;
}