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

    题目描述

    某大学有 n个职员,编号为 1n。

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

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

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

    输入格式

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

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

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

    输出格式

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

    输入输出样例

    输入 #1
    7
    1
    1
    1
    1
    1
    1
    1
    1 3
    2 3
    6 4
    7 4
    4 5
    3 5
    
    输出 #1
    5
    

    说明/提示

    数据规模与约定

    对于100% 的数据,保证  1n6×10^3,128ri127,1l,kn,且给出的关系一定是一棵树。

    很显然,这是一个树形dp(我也没学过,老师说的QWQ),那么首先要找状态转移方程,每个节点有两个需要dp的条件,一个是选或者不选,另一个是当前位置,所以开一个二维数组来记录数据。第一维表示状态,第二维记录位置。所以思路就很清晰了,详情见代码

    代码:

    #include<iostream>
    #include<cstdio>
    #include<cmath>
    #include<cstring>
    #include<cstdlib>
    #include<algorithm>
    using namespace std;
    int ru[12010],po[12010],ne[12010],f[5][12010];//开数组,ru数组的含义是有多少个父亲节点,po是父节点编号,next就是子节点,二维的f数组则记录最值答案
    int n,a,b,root;//root是根结点
    void dp(int x)
    {
    for(int i=po[x];i;i=ne[i])//我明白这里了,这条边当然要从它的起点开始循环,那么循环一次之后就要到它的下一条边
    {
    dp(i);
    f[1][x]=max(max(f[1][x],f[1][x]+f[0][i]),f[0][i]);//其实这里就是分类讨论,讨论有几种情况,这里分别是:1.只选当前的节点。2.选当前节点并且加上不选的子节点的最值。3.赋值不选的那个子节点位置上的最值。可能会有疑问,我这个大的分类就是选这个节点啊,为什么还能不选这个节点,只选子节点。我也有这个疑问。。。。。。。。。。。。。。。。。。
    f[0][x]=max(max(f[0][x],f[0][x]+f[1][i]),max(f[1][i],f[0][i]));//这里多了一种情况:1.自己不选。2.自己不选选子节点。3.只选子节点。4.不选子节点。我也晕了
    }
    }
    int main()
    {
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&f[1][i]);
    for(int i=1;i<=n;i++)
    {
    scanf("%d%d",&b,&a);
    ru[b]++;
    ne[b]=po[a];
    po[a]=b;
    }
    for(int i=1;i<=n;i++)
    {
    if(ru[i]==0)//如果一个节点没有父亲节点,那肯定是根结点
    {
    root=i;//记录根结点的位置
    break;
    }
    }
    dp(root);//从根结点开始dp,因为函数里是先递归,再写内容,所以会从最下面的叶子节点开始,那么到最后数据都在根节点里
    printf("%d",max(f[1][root],f[0][root]));//同上
    return 0;
    }

  • 相关阅读:
    PyCharm 3.4注册码
    监控Oracle索引是否被启用
    Oracle execute plan 原理分析与实例分享(转)
    6 个重构方法可帮你提升代码质量(转载)
    跟我一起学Oracle 11g【2】----用户管理(转载)
    notecore设置linux/Unix系统文件权限
    Unicode特殊字符的坑
    net多线程
    走过的HttpClient坑
    Postgre备份还原
  • 原文地址:https://www.cnblogs.com/57xmz/p/12825010.html
Copyright © 2011-2022 走看看