URAL 1039 Anniversary Party

Posted Wisdom+.+

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了URAL 1039 Anniversary Party相关的知识,希望对你有一定的参考价值。

URAL 1039

思路:

树形dp

状态:dp[i][0]表示以i为根节点的子树不选取i的最大贡献

  dp[i][1]表示以i为根节点的子树选取i的最大贡献

初始状态:dp[i][0]=0(if i is a leaf)

     dp[i][1]=a[i](if i is a leaf)

状态转移:

dp[i][0]=∑max(dp[g[i][j]][0],dp[g[i][j]][1])

dp[i][1]=a[i]+∑dp[g[i][j]][0]

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=6e3+5;
vector<int>g[N];
int dp[N][2];
int a[N];
int in[N];
void dfs(int u){
    if(g[u].size()==0){
        dp[u][0]=0;
        dp[u][1]=a[u];
        return ;
    }
    for(int i=0;i<g[u].size();i++){
        dfs(g[u][i]);
    }
    for(int i=0;i<g[u].size();i++){
        dp[u][0]+=max(dp[g[u][i]][1],dp[g[u][i]][0]);
        dp[u][1]+=dp[g[u][i]][0];
    }
    dp[u][1]+=a[u];
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,u,v;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    while(cin>>u>>v){
        if(u==0&&v==0)break;
        g[v].pb(u);
        in[u]++;
    }
    int root;
    for(int i=1;i<=n;i++){
        if(in[i]==0){
            root=i;
            break;
        }
    }
    dfs(root);
    cout<<max(dp[root][0],dp[root][1])<<endl;
    return 0;
}

 

以上是关于URAL 1039 Anniversary Party的主要内容,如果未能解决你的问题,请参考以下文章

ural 1039 树dp

BestCoder 1st Anniversary ($) Hidden String(深搜)

HDU 1520 Anniversary party

POJ 2342?Anniversary party

HDU 1520 Anniversary party

Anniversary party