傳送門
文章目錄
題意：
懶得寫了，直接貼圖了。
思路：
遇事不決畫成圖，考慮將iii向p[i]p[i]p[i]連一個邊，可以發現每個點入度為111，出度為111，所以畫出來是若干個環，比如說樣例三畫出來就是這樣的(這里只畫了一個環)
上圖為p1p^1p1的情況，進而可以發現，當ppp的冪次為kkk的時候，就是將每個數與其往下數kkk個數之間連邊，比如p2p^2p2的圖如下所示：
所以我們假設環的長度為lenlenlen，那么可以發現只有lenmodk=0len\bmod k=0lenmodk=0的kkk才有意義，其他的連邊方式都是kkk的某種情況，所以我們只需要枚舉lenlenlen的因子，讓后判斷posmoddiverpos\bmod diverposmoddiver相同的位置顏色是否一樣就好了，取一個最小的因子即可。
這里寫的麻煩了，完全可以用篩法替代我這個垃圾寫法。
復雜度O(nn)O(n\sqrt n)O(nn?)。
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std
;
typedef long long LL
;
typedef unsigned long long ULL
;
typedef pair
<int,int> PII
;const int N
=200010,mod
=1e9+7,INF
=0x3f3f3f3f;
const double eps
=1e-6;int n
;
int a
[N
],p
[N
],c
[N
],col
[N
],cnt
[N
];
bool st
[N
];
vector
<int>v
,diver
;int solve() {int len
=v
.size();diver
.clear();for(int i
=1;i
<=len
/i
;i
++) {if(len
%i
==0) {diver
.pb(i
);if(1ll*i
*i
!=len
) diver
.pb(len
/i
);}}int ans
=INF
;for(auto x
:diver
) {for(int i
=0;i
<v
.size();i
++) cnt
[i
%x
]=len
/x
;for(int i
=0;i
<v
.size();i
++) {if(col
[i
%x
]==-1) col
[i
%x
]=c
[v
[i
]];else if(col
[i
%x
]!=c
[v
[i
]]) cnt
[i
%x
]--;}for(int i
=0;i
<v
.size();i
++) if(cnt
[i
%x
]==len
/x
) ans
=min(ans
,x
);for(int i
=0;i
<v
.size();i
++) col
[i
%x
]=-1;}return ans
;
}int main()
{
memset(col
,-1,sizeof(col
));int _
; scanf("%d",&_
);while(_
--) {scanf("%d",&n
);for(int i
=1;i
<=n
;i
++) scanf("%d",&p
[i
]),st
[i
]=0;for(int i
=1;i
<=n
;i
++) scanf("%d",&c
[i
]);int ans
=INF
;for(int i
=1;i
<=n
;i
++) if(!st
[i
]&&!st
[p
[i
]]) {if(i
==p
[i
]) {ans
=1; break;}v
.clear();v
.pb(i
); v
.pb(p
[i
]); st
[i
]=1; st
[p
[i
]]=1;int now
=p
[p
[i
]];while(!st
[now
]) {v
.pb(now
);st
[now
]=1;now
=p
[now
];}int len
=v
.size();ans
=min(ans
,solve());}printf("%d\n",ans
);}return 0;
}
總結
以上是生活随笔為你收集整理的Educational Codeforces Round 84 (Rated for Div. 2) D. Infinite Path 构建环 + 思维的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
