SGU 195 New Year Bonus Grant(트리 dp)

제목: 나무 한 그루를 주고 뿌리 노드는 1이며 다음과 같은 요구에 따라 가장 많은 노드를 선택해야 한다.
① 1 선택 불가
② 현재 노드를 선택하면 해당 노드의 부모 노드와 아들이 선택할 수 없습니다.
③ 어떤 노드의 어떤 아들 노드가 선택되면 그 노드의 다른 아들은 선택될 수 없다.
사고방식: 비교적 간단한 트리 dp로 반나절 동안 문제를 이해하지 못했는데 다 쓴 후에 표칭 달리기 데이터와 문제의 뜻을 잘못 이해한 것을 발견했다. dp[u][2]로 u 노드가 선택된 최대치를 표시하면 dp[u][1]=sum{dp[v][0]}+1, dp[u][0]=max(sum{dp[v][0]]]]]]]],sum{dp[v][0]}-min{dp[v][0]+dp[v]]]]]]]);경로를 마지막으로 다시 한 번 뒤졌으면 좋겠어요...
코드:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=500000+10;
struct Edge
{
    int v,next;
}edges[maxn];
int head[maxn],dp[maxn][2],flag[maxn],nEdge;
int ans[maxn],cnt;
void AddEdge(int u,int v)
{
    nEdge++;
    edges[nEdge].v=v;
    edges[nEdge].next=head[u];
    head[u]=nEdge;
}
void dfs(int u)
{
    dp[u][0]=0;dp[u][1]=1;
    int sum=0,tmp=inf;
    for(int k=head[u];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        dfs(v);
        dp[u][1]+=dp[v][0];
        sum+=dp[v][0];
        if(dp[v][0]-dp[v][1]<tmp)
        {
            tmp=dp[v][0]-dp[v][1];
            flag[u]=v;
        }
    }
    dp[u][0]=max(sum,sum-tmp);
}
void dfs2(int u,int s)
{
    for(int k=head[u];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        if(s==1) dfs2(v,0);
        else
        {
            if(flag[u]!=v)
              dfs2(v,0);
            else
            {
                ans[cnt++]=v;
                dfs2(v,1);
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n;
    scanf("%d",&n);
    memset(head,0xff,sizeof(head));
    nEdge=-1;
    int v;
    for(int i=2;i<=n;++i)
    {
        scanf("%d",&v);
        AddEdge(v,i);
    }
    dfs(1);
    printf("%d
",dp[1][0]*1000);
    cnt=0;
    dfs2(1,0);
    sort(ans,ans+cnt);
    for(int i=0;i<cnt;++i)
    {
        if(i) printf(" ");
        printf("%d",ans[i]);
    }
    printf("
");
    return 0;
}