洛谷 P1352 没有上司的舞会（树形dp）

传送门

---

解题思路

就是一道树形dp的模板。

偷得课件QAQ：

这个题用dp[i][0]表示i不去时以i为根的子树的最大快乐指数，

　　　　dp[i][1]表示i去时以i为根的子树的最大快乐指数。

然后转移方程就很容易写出来了。见代码。

最后要注意输入的u,v是v是u的上司。

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,r[6005],cnt,dp[6005][2],p[6005],root;
bool vis[6005];
struct node{
    int v,next;
}e[6005];
void insert(int u,int v){
    cnt++;
    e[cnt].v=v;
    e[cnt].next=p[u];
    p[u]=cnt; 
}
void dfs(int u){
    for(int i=p[u];i!=-1;i=e[i].next){
        int v=e[i].v;
        dfs(v);
        dp[u][0]+=max(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][0];
    }
}
int main(){
    memset(p,-1,sizeof(p)); 
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>dp[i][1];
    }
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        vis[u]=1;
        insert(v,u);
    }
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            root=i;
            break;
        }
    }
    dfs(root);
    cout<<max(dp[root][0],dp[root][1]);
    return 0;
}