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;
}

 

上一篇:VT在BIOS内已开启,但CPU-v显示未开启


下一篇:四级英语day8