#树形dp#洛谷 1352 jzoj 1662 codevs 1380 没有上司的舞会

题目

在一棵关系树上，父节点是子节点的上司，如果有子节点，那么不可以有它们的父节点，每个节点都有值，求选择节点的最大值。

---

分析

树形dp，首先找到根，有两种可能
$f[x][0]表示不选节点x的最大值，f[x][1]表示选节点x的最大值（初始值就是节点的值）$
容易得到$f[x][0]+=max(f[son][0],f[son][1])，f[x][1]+=f[son][0]$

---

代码

#include <cstdio>
#include <cctype>
using namespace std;
struct node{int y,next;}e[6001];
int f[6001][2],ls[6001],n; bool v[6001];
int in(){
    int ans=0,f=1; char c=getchar();
    while (!isdigit(c)&&c!='-') c=getchar();
    if (c=='-') f=-f,c=getchar();
    while (isdigit(c)) ans=ans*10+c-48,c=getchar();
    return ans*f;
}
int max(int a,int b){return (a>b)?a:b;}
void dp(int x){
    for (int i=ls[x];i;i=e[i].next){
        dp(e[i].y);
        f[x][0]+=max(f[e[i].y][0],f[e[i].y][1]);
        f[x][1]+=f[e[i].y][0];
    }
}
int main(){
    n=in();
    for (int i=1;i<=n;i++) f[i][1]=in();
    for (int i=1;i<n;i++){
        int x,y;
        x=in(); y=in(); v[x]=1;//标记为子节点
        e[i]=(node){x,ls[y]}; ls[y]=i;//连边（成员连上司）
    }
    int root; for (root=1;root<=n&&v[root];root++);//找根
    dp(root); return !printf("%d",max(f[root][0],f[root][1]));//可能不选根节点，也可能选
}