codeforces 1327D Infinite Path 图论+思维

https://vjudge.net/problem/CodeForces-1327D

题目大意：大概就是说找一个最小的$k$，对序列$p$进行转换之后，存在一个起始位置$i$使得$c[i]=c[p[i]]=c[p[p[i]]]\dots \dots$(一条无限长的路径)。

思路：首先有一个结论就是，对于一个$n$的全排列$p$，我们把它当成一个有$n$个顶点的图，给$(i,p[i])$连一条无向边，那么得到的图一定是由几个独立的环组成的(自环也算)。所以依次在每个环上找最小的$k_i$，最后总体取个最小值就是答案了。我们假设某个环有$m$个元素组成，设为$cycle[1\dots m]$，依据定义有：$$p[cycl{e}_i]=cycl{e}_{i+1modm}$$$$p^2[cycl{e}_i]=p[p[cycl{e}_i]]=cycl{e}_{i+2}modm$$$$\dots \dots$$$$p^k[cycl{e}_i]=cycl{e}_{i+k}modm$$所以对每个环来说，我们就是要在这个环上找到一条步长最小的无限长的路径。

那么对于步长$k$来说，循环节的长度就是$m/gcd(k,m)$，那么对于$gcd(k_1,m)=gcd(k_2,m)$的情况来说，其实只需要验证其中$1$个就行了，后者和前者仅仅是顺序不同而已。所以验证可以整除$m$的步长就行了。也就是$m$的因子。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;

const int maxn=2e5+5;

int t,n,len,ans;
int p[maxn],c[maxn],cycle[maxn],vis[maxn];

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&p[i]);
        for(int i=1;i<=n;i++)
            scanf("%d",&c[i]);
        ans=INF;
        for(int pos=1;pos<=n;pos++)//枚举开始位置
        {
            if(vis[pos])
                continue;
            len=0;
            int v=pos;
            while(!vis[v]) //找环
            {
                cycle[++len]=v;
                vis[v]=1;
                v=p[v];
            }
            bool flag=1;
            for(int i=1;i<=len;i++)
            {
                if(len%i)
                    continue; //必须整除
                for(int j=1;j<=i;j++)//步长为i 起始位置为j
                {
                    flag=1;
                    for(int k=j;k+i<=len;k+=i)
                    {
                        if(c[cycle[k]]!=c[cycle[k+i]])
                        {
                            flag=0;
                            break;
                        }
                    }
                    if(flag)
                    {
                        ans=min(ans,i);
                        break;
                    }
                }
                if(flag)
                    break;
            }
        }
        printf("%d\n",ans);
        memset(vis,0,sizeof(int)*(n+1));
    }
    return 0;
}