hdu1520 Anniversary party 【树形dp】

链接：http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意：给你一棵树，每个点都有权值，现在要你选一些点，这些点与父亲不能同时选，问权值最大多少。

分析：树形dp入门题，直接用dp[i][0/1]表示第i个节点选或不选，转移：

dp[u][0]+=max(dp[v][0],dp[v][1])

dp[u][1]+=dp[v][0]

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 6010
#define Mm 12005
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
struct edge {
    int v,next;
}e[Mm];
int tot,head[Mn];
void addedge(int u,int v) {
    e[tot].v=v;
    e[tot].next=head[u];
    head[u]=tot++;
}
int n;
void init() {
    CLR(head,-1);
    tot=0;
}
int a[Mn];
int dp[Mn][2];
void dfs(int u,int fa) {
    dp[u][1]=a[u];
    dp[u][0]=0;
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        if(v==fa) continue;
        dfs(v,u);
        dp[u][1]+=max(0,dp[v][0]);
        dp[u][0]+=max(dp[v][0],max(0,dp[v][1]));
    }
}
int main() {
    while(~scanf("%d",&n)) {
        for(int i=1;i<=n;i++) {
            scanf("%d",&a[i]);
        }
        int u,v;
        init();
        while(1){
            scanf("%d%d",&u,&v);
            if(u==0&&v==0) break;
            addedge(u,v);
            addedge(v,u);
        }
        dfs(1,-1);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}