天天看點

codeforces 115A Party dfs

題意:給定一片森林,對點進行分組,每一組的點必須沒有父子關系。

做法:直接DFS,記錄每個點在各自數中的深度,最大的深度值極為答案。

#include <iostream>
#include <cstdio>
#include <cstring>
#define LMT 2002
using namespace std;
int ans,n;
int father[LMT],is[LMT],next[LMT],all;
struct line
{
    int u,v,next;
}e[LMT];
void insert(int u,int v)
{
    e[all].u=u;
    e[all].v=v;
    e[all].next=next[u];
    next[u]=all++;
}
void dfs(int u,int lev)
{
    if(ans<lev)ans=lev;
    for(int x=next[u];x!=-1;x=e[x].next)
    dfs(e[x].v,lev+1);
}
int main()
{
    int p;
    scanf("%d",&n);
    memset(next,-1,sizeof(next));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&p);
        if(p==-1)is[i]=1;
        else insert(p,i);
    }
    for(int i=1;i<=n;i++)
    if(is[i])dfs(i,1);
    printf("%d\n",ans);
    return 0;
}