sgu 195 New Year Bonus Grant 树形dp

       给一棵树，根节点为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;
}