旅行计划 51Nod - 1273

https://www.51nod.com/Challenge/Problem.html#!#problemId=1273

 

最长的那条链肯定是题中第一次要走的那条路经 第二次要走的就是拿掉这条链后第二长的链

按子树中节点的最大深度来划分轻重链 然后求一下每条链的长度 排个序即可

辣鸡51nod 第一个样例WA到死 打开发现是 "1 0\n 0" 这他妈是个鬼样例啊 需要特判

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;

struct node1
{
    int v,next;
};

struct node2
{
    int id,val;
};

node1 edge[2*maxn];
node2 ary[maxn];
int first[maxn],fa[maxn],deep[maxn],gou[maxn],son[maxn],top[maxn];
int n,s,num;

bool cmp(node2 n1,node2 n2)
{
    if(n1.val==n2.val) return n1.id<n2.id;
    else return n1.val>n2.val;
}

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfsI(int cur)
{
    int i,v;
    son[cur]=-1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]){
            fa[v]=cur,deep[v]=deep[cur]+1;
            dfsI(v);
            if(son[cur]==-1||deep[gou[cur]]<deep[gou[v]]||(deep[gou[cur]]==deep[gou[v]]&&gou[cur]>gou[v])){
                son[cur]=v,gou[cur]=gou[v];
            }
        }
    }
    if(son[cur]==-1) gou[cur]=cur;
}

void dfsII(int cur,int tp)
{
    int i,v;
    top[cur]=tp;
    if(son[cur]==-1) return;
    dfsII(son[cur],tp);
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa[cur]&&v!=son[cur]){
            dfsII(v,v);
        }
    }
}

int main()
{
    int i,u,v,tar;
    scanf("%d%d",&n,&s);
    if(n==1&&s==0){
        scanf("%d",&u);
        printf("0\n");
        return 0;
    }
    s++;
    memset(first,-1,sizeof(first));
    num=0;
    for(u=2;u<=n;u++){
        scanf("%d",&v);
        v++;
        addedge(u,v),addedge(v,u);
    }
    dfsI(s);
    dfsII(s,s);

    num=0;
    for(i=1;i<=n;i++){
        if(son[i]==-1){
            num++;
            ary[num].id=i,ary[num].val=0;
            u=i,tar=top[u];
            while(1){
                ary[num].val++;
                if(u==tar) break;
                u=fa[u];
            }
        }
    }
    sort(ary+1,ary+num+1,cmp);

    printf("%d\n",s-1);
    for(i=1;i<=num;i++) printf("%d\n",ary[i].id-1);
    return 0;
}