L2-026 小字辈（并查集路劲压缩思路）

这道题呢在去年比赛的时候还写不出来只能得一点点的分数但是现在已经一会就能a掉了

果然进步了鸭，用类似路径压缩的方法标记长度这样就能剪掉很多时间

#include <iostream>
#include <vector>
using namespace std;
const int MAXN=111111;

int num[MAXN];
int ans[MAXN];
int vis[MAXN];

vector<int> vt[MAXN];
int find(int x)
{
    if(vis[x])
        return ans[x];
    vis[x]=1;
    if(x!=-1) 
        ans[x]+=find(num[x]);
    return ans[x];
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<MAXN;i++)
    {
        num[i]=i;
        ans[i]=1;
    }
    for(int i=1;i<=n;i++)
    {
        int k;
        scanf("%d",&k);
        num[i]=k;
    }
    int max1=-1;
    for(int i=1;i<=n;i++)
    {
        int kk=find(i);
        if(kk>max1)
        {
            max1=kk;
            vt[max1].push_back(i);
        }
        else if(kk==max1)
            vt[max1].push_back(i);
    }
    printf("%d\n",max1);
    for(int i=0;i<vt[max1].size();i++)
    {
        if(!i) printf("%d",vt[max1][i]);
        else printf(" %d",vt[max1][i]);
    }
    return 0;
}