J - 最大集合 - 并查集

J - 最大集合 - 并查集

题干

给定一个1−N的排列 A[1], A[2], … A[N]，定义集合 S[K] = {A[K], A[A[K]], A[A[A[K]]] … }。
显然对于任意的 K=1…，S[K]都是有限集合。你能求出其中包含整数最多的 S[K]的大小吗？

输入格式
第一行包含一个整数 N(1≤N≤100000)

第二行包含 N 个两两不同的整数，A[1], A[2], … A[N]。(1≤A[i]≤N)
输出格式最大的 S[k]的大小。

Sample Input
7
6 5 1 4 2 7 3

Sample Output
4

知识点&算法

题意即:
定义S[k]为从A[K]开始，不断将A[K]的值做下标，获得新的值，然后作为新的下标再获得新的值，如此反复得到的全部A中元素的集合，然后找一个元素最多的S[k]。
举个例子手算一遍，发现集合是闭合的，相当于考察连通性，求最大连通分支的元素个数。
直接上并查集，单独开一个数组统计连通分支的元素个数。
特别要注意的是，这里的元素下标是从1开始的。

题解

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int n,t,ans,anc[100005],cnt[100005];

int findAnc(int n){
    if(anc[n] == n) return n;
    while(anc[n] != n) n = anc[n];
    return n;
}

int main()
{
    cin>>n;
    for(int i = 1 ; i <= n ; ++i) cnt[i] = 1;
    for(int i = 1 ; i <= n ; ++i) anc[i] = i;
    for(int i = 1 ; i <= n ; ++i){
        scanf("%d",&t);
        if(findAnc(i) != findAnc(t)){
            cnt[findAnc(i)] += cnt[findAnc(t)];
            anc[findAnc(t)] = findAnc(i);
        }
        ans = max(ans,cnt[findAnc(i)]);
    }
    cout<<ans;
    return 0;
}