前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >hdoj 1520 Anniversary party(树形dp)

hdoj 1520 Anniversary party(树形dp)

作者头像
xindoo
发布2021-01-22 15:21:32
2880
发布2021-01-22 15:21:32
举报
文章被收录于专栏:XINDOO的专栏

题目链接

按照等级我们可以建一颗树,如图

我们可以把一个节点当做一个人,每个节点都有一个权重。按照题目意思,如果我们取了某个节点,那么他的父节点和子节点都是不能取的。按要求选取节点,使得选取节点的权重和最大。

DP,用no表示不选择i点时,i点及其子树能选出的最多人数,is表示选择i点时,i点及其子树的权值和最大。

状态转移方程:

—对于叶子节点 dp[k].no = 0, dp[k].no = a[k].v

—对于非叶子节点i,

—dp[i].no = ∑max(dp[j][0], dp[j][1]) (j是i的儿子)

—dp[i].is = v + ∑dp[j].no (j是i的儿子)

代码:

代码语言:javascript
复制
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string.h>
using namespace std;

struct node
{
    int v;
    int is, no;
    vector<int> son;
}a[6010];

bool f[6010];

void dp(int x)
{
    int t;
    for (unsigned int i = 0; i < a[x].son.size(); i++)
    {
        t = a[x].son[i];
        dp(t);
        a[x].is += a[t].no;
        a[x].no += max(a[t].no, a[t].is);
    }
}

int main()
{
    int n;
    while (scanf("%d", &n) != EOF)
    {
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i].v);
            a[i].son.clear();
            a[i].is = a[i].v;
            a[i].no = 0;
        }
        int son, fa;
        memset(f, 0, sizeof(f));
        while (scanf("%d %d",&son, &fa) && son+fa)
        {
            a[fa].son.push_back(son);
            f[son] = 1;
        }
        for (int i = 1; i <= n; i++)
        {
            if (f[i] == false)
            {
                dp(i);
                printf("%d\n",max(a[i].is, a[i].no));
                break;
            }
        }
    }
    return 0;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2013-04-26,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档