zoukankan      html  css  js  c++  java
  • P1352 没有上司的舞会(树形DP)

    题目描述

    某大学有 nn 个职员,编号为 1ldots n1n。

    他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。

    现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 r_iri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。

    所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。

    输入格式

    输入的第一行是一个整数 nn。

    第 22 到第 (n + 1)(n+1) 行,每行一个整数,第 (i+1)(i+1) 行的整数表示 ii 号职员的快乐指数 r_iri

    第 (n + 2)(n+2) 到第 2n2n 行,每行输入一对整数 l, kl,k,代表 kk 是 ll 的直接上司。

    输出格式

    输出一行一个整数代表最大的快乐指数。

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=2e5+100;
    int n;
    vector<int> g[maxn];
    int dp[2][maxn];
    //dp[0]表示不选择该节点的最优解
    //dp[1]表示选择该节点的最优解
    int a[maxn];
    void dfs (int u,int pre) {
        dp[0][u]=0;
        dp[1][u]=a[u];
        for (int v:g[u]) {
            if (v==pre) continue;
            dfs(v,u);
            dp[0][u]+=max(dp[1][v],dp[0][v]);
            dp[1][u]+=dp[0][v];
        }
    } 
    int main () {
        scanf("%d",&n);
        for (int i=1;i<=n;i++) scanf("%d",a+i);
        for (int i=1;i<n;i++) {
            int x,y;
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        dfs(1,0);
        printf("%d
    ",max(dp[0][1],dp[1][1]));
    }
  • 相关阅读:
    Python MySQL(SQL语句的补充3)
    Python MySQL(pymysql的使用day1)
    Python MySQL(SQL语句的补充2)
    Python MySQL(对外键使用方法的补充)
    Python MySQL(学习SQL语句)
    Python MySQL数据库的基本使用
    关于http响应内容压缩的一点小积累。
    同一账号,后一用户登录,前一个用户则被踢掉
    对序列化的学习。
    MySQL 函数积累
  • 原文地址:https://www.cnblogs.com/zhanglichen/p/13466358.html
Copyright © 2011-2022 走看看