关闭

sgu 195 New Year Bonus Grant 树形dp

490人阅读 评论(0) 收藏 举报
分类:

       给一棵树,根节点为1,对于根节点以外的每一个节点,可以点亮该节点或他的一个儿子节点。求最多点亮的节点数*1000,以及一种可行的点亮方案。

       记dp[i][0]为i不点时,i的子树中点亮的节点数,dp[i][1]为i点亮时,i的子树中点亮的节点数。初始状态除了1意外所有的dp[i][1]都是1,然后从根节点开始dfs,dp[i][1]=dp[i][1]+sum(dp[v][0],g[i][v]==true),记res=sum(dp[v][0],g[i][v]==true),dp[i][0]=max(res-dp[v][0]+dp[v][1],g[i][v])。方案的话另开一个bool数组,更新Dp[i][0]的时候记录一下他的子节点的1,0情况就可以。

     

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
int n,m;
const int maxn=500000+500;
int dp[maxn][2];
bool ok[maxn];
vector<int> g[maxn];
void dfs(int i)
{
    int res=0;
    for (int j=0; j<g[i].size(); j++)
    {
        int v=g[i][j];
        dfs(v);
        res+=dp[v][0];
    }
    dp[i][1]+=res;
    int ans=res;
    int k=-1;
    for (int j=0; j<g[i].size(); j++)
    {
        int v=g[i][j];
        if (ans<res-dp[v][0]+dp[v][1])
        {
            k=v;
            ans=res-dp[v][0]+dp[v][1];
        }
    }
    dp[i][0]=ans;
    if (k!=-1) ok[k]=true;
}
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    memset(dp,0,sizeof dp);
    for (int i=1; i<=n; i++)
    dp[i][1]=1,g[i].clear();

    for (int i=2; i<=n; i++)
    {
        scanf("%d",&m);
        g[m].push_back(i);
    }
    memset(ok,0,sizeof ok);
    dfs(1);

    cout<<dp[1][0]*1000<<endl;
    for (int i=1; i<=n; i++)
    if (ok[i]) cout<<i<<" ";
    cout<<endl;
    return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:136907次
    • 积分:3772
    • 等级:
    • 排名:第8832名
    • 原创:250篇
    • 转载:0篇
    • 译文:0篇
    • 评论:35条
    文章分类
    最新评论