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;

}