(树形dp) Anniversary party

https://cn.vjudge.net/contest/264707#problem/A
要有一个庆祝乌拉尔州立大学80周年。这所大学有一个员工的层次结构。这意味着主管的关系形成一个树在特列季亚科夫校长”。为了使党的滑稽的每一个,校长不希望员工和他或她的直属主管到场。人事办公室评估每个员工的欢乐,所以每个人都有一些数(等级)连接到他或她。你的任务是与客人欢宴评级的最大金额列出客人。

题意即是员工与直属主管不能同时出席,且每个人有一个快乐值,问最大的快乐值
树形dp入门,对每一个人有两个状态,一为参加宴会,二是不参加,参加的话他的员工即不能参加,不参加的话员工能参加和不参加
多组输入!!!wa了10多次

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e4 + 5;
const int inf = 0x3f3f3f3f;
ll dp[maxn][2];
int n, in[maxn], vis[maxn];
vector<int> e[maxn];
void dfs(int u) {
    vis[u] = 1;
    for(int v : e[u]) {
        if(vis[v]) continue;
        dfs(v);
        dp[u][1] += dp[v][0];
        dp[u][0] += max(dp[v][0], dp[v][1]);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    while(cin >> n) {
        memset(dp, 0, sizeof(dp));
        memset(vis, 0, sizeof(vis));
        for(int i = 1; i <= n; i++) {
            cin >> dp[i][1];
            e[i].clear();
        }
        int u, v;
        while(cin >> u >> v) {
            if(u == 0 && v == 0) break;
            in[u]++;
            e[v].emplace_back(u);
        }
        int tag;
        for(int i = 1; i <= n; i++)
            if(in[i] == 0) {
                tag = i;
                break;
            }
        dfs(tag);
        cout << max(dp[tag][1], dp[tag][0]) << endl;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_40588429/article/details/84725059
今日推荐