[POJ2342]Anniversary party

OJ题号：POJ2342、洛谷1352

思路：树形动规（递归实现），每次返回一个pair，分别记录包括本人的最大值和不包括本人的最大值，每次如果遇到conviviality rating为负的就忽略。

#include<cstdio>
#include<vector>
#include<algorithm>
const int N=6001;
std::vector<int> c[N];
int r[N],p[N]={0};
inline void add_edge(const int a,const int b) {
    c[a].push_back(b);
    p[b]=a;
}
std::pair<int,int> dfs(const int p) {
    int w1=r[p],w2=0;
    for(unsigned int i=0;i<c[p].size();i++) {
        std::pair<int,int> t=dfs(c[p][i]);
        if(t.second>0) w1+=t.second;
        if(t.first>0) w2+=t.second?std::max(t.first,t.second):t.first;
    }
    return std::make_pair(w1,w2);
}
int main() {
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&r[i]);
    for(int i=1;i<n;i++) {
        int l,k;
        scanf("%d%d",&l,&k);
        add_edge(k,l);
    }
    int root;
    for(int i=1;i<=n;i++) {
        if(!p[i]) {
            root=i;
            break;
        }
    }
    std::pair<int,int> ans=dfs(root);
    printf("%d
",std::max(ans.first,ans.second));
    return 0;
}