結果
問題 |
No.3256 Permutation Equation
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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":" "); }