結果

問題 No.3256 Permutation Equation
ユーザー GOTKAKO
提出日時 2025-09-06 07:19:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 55 ms / 2,000 ms
コード長 1,712 bytes
コンパイル時間 2,338 ms
コンパイル使用メモリ 208,900 KB
実行使用メモリ 10,628 KB
最終ジャッジ日時 2025-09-06 07:20:04
合計ジャッジ時間 8,807 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

random_device rnd;
mt19937 mt(rnd());

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N; cin >> N;
    vector<int> P(N);
    for(auto &a : P) cin >> a,a %= N;
    
    vector<vector<vector<int>>> cycle(N+1);
    vector<bool> already(N);
    for(int i=0; i<N; i++) if(!already.at(i)){
        vector<int> now;
        int pos = P.at(i);
        while(already.at(pos) == false) now.push_back(pos),already.at(pos) = true,pos = P.at(pos);
        cycle.at(now.size()).emplace_back(now);
    }

    vector<int> Q(N);
    int inc = 0;
    for(int i=1; i<=N; i++){
        if(i%2 == 0){
            while(cycle.at(i).size()){
                auto C1 = cycle.at(i).back(); cycle.at(i).pop_back();
                if(cycle.at(i).empty()){inc = -1; break;}
                auto C2 = cycle.at(i).back(); cycle.at(i).pop_back();
                for(int k=0; k<i; k++) Q.at(C1.at((k+i-1)%i)) = C2.at(k),Q.at(C2.at(k)) = C1.at(k);
            }
            if(inc == -1) break;
        }
        else{
            for(auto &C : cycle.at(i)) for(int k=0; k<i; k++) Q.at(C.at((k+i-1)%i)) = C.at((k+inc)%i);
            inc++;
        }
    }
    if(inc == -1){cout << "No\n"; return 0;}
    for(auto &q : Q) q = (q-1+N)%N;
    for(auto &p : P) p = (p-1+N)%N;
    for(int i=0; i<N; i++){
        int p = P.at(i),q = Q.at((Q.at(i)+1)%N);
        if(p != q){
            cout << "Wrong" << endl;
            for(auto p : P) cout << p+1 << " "; cout << endl;
            for(auto p : Q) cout << p+1 << " "; cout << endl;
            return 0;
        }
    }
    cout << "Yes\n";
    for(int i=0; i<N; i++) cout << Q.at(i)+1 << (i==N-1?"\n":" ");

} 
0