CODEVS 1380没有上司的舞会

#include<stdio.h>#include<iostream>#include<vector>using namespace std;vector<int>sons[6010];int n,happiness[6010],father[6010],son,fa,f[6010][5];int mmax(int a,int b){    return a>b?a:b;}int d(int cur,int state){//遍历到cur点 state表示happiness[cur]取或不取    if(f[cur][state]>0) return f[cur][state];    if(sons[cur].size()==0) return f[cur][state]+=(state==1?happiness[cur]:0);//如果cur没有子节点 即cur为叶节点    int sum=0;    if(state==1) sum+=happiness[cur];//注意！！！happiness[cur]只加一次！！！    for(int i=0;i<sons[cur].size();i++){        int tryson=sons[cur][i];        if(state==1){//state==1 则取            sum+=d(tryson,0);        }        if(state==0){//state==0 则不取            int aa=mmax(d(tryson,0),d(tryson,1));            sum+=aa;        }    }    return f[cur][state]=sum;}int main(){    freopen("1380.in","r",stdin);    freopen("1380.out","w",stdout);    scanf("%d",&n);    for(int i=1;i<=n;i++) scanf("%d",&happiness[i]);    while(1){        scanf("%d %d",&son,&fa);        if(son==fa) break;        sons[fa].push_back(son);//将fa的son导入sons        father[son]=fa;//son的father是fa    }    for(int i=1;i<=n;i++){        if(father[i]==0){//找到根节点了            printf("%d",mmax(d(i,1),d(i,0)));        }    }    return 0;}

题解：树形DP（记忆化搜索），由于不一定是二叉树，所以我用了STL中的vector数组来储存每个节点的子节点。d(cur,state) 其中state表示cur点取或不取的最大值。