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