1362. Classmates 2

http://acm.timus.ru/problem.aspx?space=1&num=1362

水题，树形DP

代码：

#include<iostream>
#include<stack>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<string>
#include<cmath>

using namespace std;

typedef long long ll;
typedef pair<int,int> pp;
const int INF=0x3f3f3f3f;
const int N=100002;
int head[N],I;
int d[N];
priority_queue<int>qt[N];
struct node
{
    int j,next;
}edge[N*2];
void add(int i,int j)
{
    edge[I].j=j;
    edge[I].next=head[i];
    head[i]=I++;
}
int dp(int pre,int x)
{
    if(d[x]!=-1)
    return d[x];

    for(int t=head[x];t!=-1;t=edge[t].next)
    {
        int j=edge[t].j;
        if(j==pre)continue;
        qt[x].push(dp(x,j));
    }
    d[x]=0;
    int t=0;
    while(!qt[x].empty())
    {
        ++t;
        d[x]=max(d[x],t+qt[x].top());
        qt[x].pop();
    }
    return d[x];
}
int main()
{
    //freopen("data.in","r",stdin);
    int n;
    cin>>n;
    memset(head,-1,sizeof(head));I=0;
    memset(d,-1,sizeof(d));
    for(int i=1;i<=n;++i)
    {
        int k;
        while(scanf("%d",&k))
        {
            if(k==0) break;
            add(i,k);add(k,i);
        }
    }
    int root;
    cin>>root;
    cout<<dp(-1,root)<<endl;
    return 0;
}