51Nod 2605 没有上司的舞会

2021-07-26 03:00:06 阅读：250 来源： 互联网

标签：2605 舞会 51Nod cnt int edge now head dp

题目大意：

题解：
树形\(dp\)很经典的一道题，各职员的上下级关系可以构成一棵树。
设\(dp[i][1/0]\)表示编号为\(i\)的职员参加或不参加聚会时，以他为根的子树所能获得的最大快乐指数。
当某职员参加聚会时，他的下级只能不参加；当他不参加聚会时，他的下级可以选择参加或不参加。
状态转移方程：

\[dp[fa][0] = \sum max\{dp[son][0], dp[son][1]\} \]

\[dp[fa][1] = r[fa] + \sum dp[son][0] \]

答案为：\(max\{dp[1][0], dp[1][1]\}\)。

#include <cstring>
#include <iostream>
using namespace std;

int dp[100010][2], n, r[100010], cnt, head[100010];
struct Edge {
    int v, next;
} edge[200010];

void addEdge(int u, int v) {
    edge[++cnt].v = v;
    edge[cnt].next = head[u];
    head[u] = cnt;
}

void dfs(int now, int pre) {
    dp[now][0] = 0;
    dp[now][1] = r[now];
    for (int i = head[now]; i; i = edge[i].next) {
        if (edge[i].v == pre) {
            continue;
        }
        dfs(edge[i].v, now);
        dp[now][0] += max(dp[edge[i].v][0], dp[edge[i].v][1]);
        dp[now][1] += dp[edge[i].v][0];
    }
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> r[i];
    }
    for (int i = 1, u, v; i < n; ++i) {
        cin >> u >> v;
        addEdge(u, v);
        addEdge(v, u);
    }
    dfs(1, 0);
    cout << max(dp[1][0], dp[1][1]);
    return 0;
}

标签：2605,舞会,51Nod,cnt,int,edge,now,head,dp
来源： https://www.cnblogs.com/IzumiSagiri/p/15059659.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

51Nod 2605 没有上司的舞会