edu div.2 D. Infinite Path 置换，什么的

D. Infinite Path

题目地址
看的神仙题解

绕口的题意

给一个排列P，和一个颜色C数组
定义两个排列的乘积，a * b = c 那么 c[i] = b[a[i]] ，
p的k次方就是 p * p * p * p * p
问k 最小是多少的时候可以有一个环上面的颜色一样。
看不懂?
来个样例
7 4 5 6 1 8 3 2 p的一次方
3 6 1 8 7 2 5 4 p的二次方
5 8 7 2 3 4 1 6 p的三次方
颜色是：
5 3 6 4 7 5 8 4
可以看出p的二次方里有个环 8 4 他的颜色都是4.所以答案是2

bb赖赖了

单纯为了理思路写博客。
首先这个排列一定是有好多环构成的。
由i -> p[i] 连一条边，那么就可以得到好多环。
p的一次方就是 i 在这个环里走一次的数字。
二次方就是走两次……
所以现在问题就是 每个点走k次，有没有个环并且他们的颜色一样。
这个怎么判断？
就是一个点开始，在自己的环里走，每次走k步，最后回到自己。并且这些颜色得一样。
于是，就可以在每个环里枚举一下环的因子，再枚举一下从哪个点开始，然后一次走k步判断一下就好了。

#include<algorithm>
#include<iostream>
#include <cstdio>
#include <string>
#include <queue>
#include <cstring>
#include <stack>
#include <map>
#include <bitset>
#include <set>
#include <unordered_set>
#include <climits>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef pair<double,double> pdd;
typedef unsigned long long ull;
typedef unordered_set<int>::iterator sit;
#define st first
#define sd second
#define mkp make_pair
#define pb push_back
void tempwj(){freopen("hash.in","r",stdin);freopen("hash.out","w",stdout);}
ll gcd(ll a,ll b){return b == 0 ? a : gcd(b,a % b);}
ll qpow(ll a,ll b,ll mod){a %= mod;ll ans = 1;while(b > 0){if(b & 1)ans = ans * a % mod;a = a * a % mod;b >>= 1;}return ans;}
struct cmp{bool operator()(const pii & a, const pii & b){return a.second > b.second;}};
int lb(int x){return  x & -x;}
//friend bool operator < (Node a,Node b)   重载
// P4551
const int inf = INT_MAX;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9+7;
const int maxn = 2e5+10;
const int M = 1e6+10;
int vis[maxn];
int c[maxn];
std::vector<int> vv[maxn];
int slen;
int temp[maxn];
int n;
void dfs(int x)
{
	vis[x] = 1;
	slen ++ ;
	temp[slen] = x;
	for(int i = 0; i < vv[x].size(); i ++ )
	{
		int v = vv[x][i];
		if(vis[v])
			continue;
		dfs(v);
	}
}
int solve(int k)// 跳k步
{
	for (int i = 1; i <= k; i ++ )
	{
		int f = 1;
		for (int j = i + k; j <= slen; j += k)
		{
			if(c[temp[i]] != c[temp[j]])
			{
				f = 0;
				break;
			}
		}
		if(f)
			return k;
	}
	return n;
}

int main()
{
	int T;
	scanf("%d",&T);
	while(T -- )
	{
		
		scanf("%d",&n);
		for (int  i= 1; i <= n; i ++ )
		{
			vv[i].clear();
			vis[i] = 0;
		}
		for (int  i= 1; i <= n; i ++)
		{
			int x;
			scanf("%d",&x);
			vv[i].pb(x);
		}
		for (int  i= 1; i <= n; i ++ )
			scanf("%d",&c[i]);
		int ans = n;
		for (int i =1 ; i <= n; i ++ )
		{
			if(vis[i] == 0)
			{
				slen = 0;
				dfs(i);
				// printf("%d %d\n",i,slen);
				for (int j = 1; j * j <= slen; j ++ )
				{
					if(slen % j == 0)
					{
						ans = min(ans,min(solve(j),solve(slen / j)));
					}
				}
			}
		}
		printf("%d\n",ans);
	}
}