HDU1520 Anniversary party(树形DP)

题意:

某公司要举办一次晚会,但是为了使得晚会的气氛更加活跃,每个参加晚会的人都不希望在晚会中见到他的直接上司,现在已知每个人的活跃指数和上司关系(当然不可能存在环),求邀请哪些人(多少人)来能使得晚会的总活跃指数最大。

思路:

当i来时,dp[i][1]+=dp[j][0]  (j为i的下属)

当i不来时,dp[i][0]+=max(dp[j][0],dp[j][1])

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=10000;
const double eps=1e-8;
const double PI = acos(-1.0);
int father[maxn],tot;
int n,a[maxn],dp[maxn][2],vis[maxn];
void init()
{
    memset(father,0,sizeof(father));
    memset(dp,0,sizeof(dp));
    memset(vis,0,sizeof(vis));
}
void dfs(int u)
{
    vis[u]=1;
    for(int i=1; i<=n; i++)
    {
        if(father[i]==u&&!vis[i])
        {
            dfs(i);
            dp[u][0]+=max(dp[i][0],dp[i][1]);
            dp[u][1]+=dp[i][0];
        }
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    while(cin>>n)
    {
        init();
        for(int i=1; i<=n; i++)
        {
            cin>>dp[i][1];
        }
        int u,v,root=1;
        while(cin>>u>>v)
        {
            if(!u&&!v)
                break;
            father[u]=v;
        }
        while(father[root])
            root=father[root];
//   cout<<root<<endl;
        dfs(root);
        cout<< max(dp[root][0],dp[root][1])<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Dilly__dally/article/details/82228813