結果

問題 No.3441 Sort Permutation 2
コンテスト
ユーザー ZeriToki
提出日時 2026-02-06 23:45:59
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 146 ms / 2,000 ms
コード長 1,558 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,613 ms
コンパイル使用メモリ 336,344 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-02-06 23:46:09
合計ジャッジ時間 9,562 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rrep(i, s, n) for (int i = (s); i < (int)(n); i++)
const long long mod=998244353;
const long long mod2=469762049;
const long long mod100=1000000007;
ll GCD(ll A,ll B){//GCD(A,B)
    while(A>=1 && B>=1){
        if(A>=B) A=(A%B);
        else B=(B%A);
    }
    if(A!=0) return A;
    return B;
}
int N;
int P[200009];
bool already[200009];
int ans[200009];
void solve(vector<int>&cycle){
    if(cycle.size()==1) return;
    int gcd=-1;
    cycle.push_back(cycle[0]);
    for(int i=1;i<(int)(cycle.size());i++){
        int d=abs(cycle[i]-cycle[i-1]);
        if(gcd==-1) gcd=d;
        else gcd=GCD(gcd,d);
    }
    for(int i=1;i*i<=gcd;i++){
        if(gcd%i==0){
            int a=i,b=gcd/i;
            if(a==b){
                ans[a]+=cycle.size()-2;
            }
            else{
                ans[a]+=cycle.size()-2;
                ans[b]+=cycle.size()-2;
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N;
    for(int i=1;i<=N;i++) cin>>P[i];
    for(int i=1;i<=N;i++){
        already[i]=false;
        ans[i]=0;
    }
    vector<int>cycle;
    for(int i=1;i<=N;i++){
        if(already[i]) continue;
        cycle.clear();
        int p=i;
        while(true){
            if(already[p]) break;
            already[p]=true;
            cycle.push_back(p);
            p=P[p];
        }
        solve(cycle);

    }
    for(int i=1;i<N;i++) cout<<ans[i]<<"\n";
}
0