結果
問題 |
No.3256 Permutation Equation
|
ユーザー |
|
提出日時 | 2025-09-06 07:03:54 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 4,042 bytes |
コンパイル時間 | 2,380 ms |
コンパイル使用メモリ | 208,024 KB |
実行使用メモリ | 10,656 KB |
最終ジャッジ日時 | 2025-09-06 07:04:07 |
合計ジャッジ時間 | 12,256 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 WA * 23 |
ソースコード
#include <bits/stdc++.h> using namespace std; template<typename T> bool chmin(T &a,T b){ //a>bならa=bに更新してtrue. if(a > b){a = b; return true;} else return false; } template<typename T> bool chmax(T &a,T b){ //a<bならa=bに更新してtrue. if(a < b){a = b; return true;} else return false; } template<typename T> T safemod(T a,T m){a %= m,a += m;return a>=m?a-m:a;} //return x = a mod m. template<typename T> T floor(T a,T b){ //return a/b切り下げ. if(b < 0) a *= -1,b *= -1; return a<0?(a+1)/b-1:a/b; } template<typename T> T ceil(T a,T b){ //return a/b切り上げ. if(b < 0) a *= -1,b *= -1; return a>0?(a-1)/b+1:a/b; } template<typename T> pair<T,T> invgcd(T a,T b){ //return {gcd(a,b),x} (xa≡g(mod b)) a = safemod(a,b); if(a == 0) return {b,0}; T x = 0,y = 1,memob = b; while(a){ T q = b/a; b -= a*q; swap(x,y); y -= q*x; swap(a,b); } if(x < 0) x += memob/b; return {b,x}; } template<typename T> bool isABmoreC(T a,T b,T c){return a>ceil(c,b);} //a*b=cはfalse a*b>c -> a>ceil(c/b)?. template<typename T> bool isABmoreC2(T a,T b,T c){return a>=ceil(c,b);} //a*b=cはtrue. template<typename T> bool isABlessC(T a,T b,T c){return a<floor(c,b);} //a*b=cはfalse. template<typename T> bool isABlessC2(T a,T b,T c){return a<=floor(c,b);} //a*b=cはtrue. template<typename T> T Kthpower(T a,T k){ //return a^k オーバーフローは考慮しない. T ret = 1; while(k){ if(k&1) ret *= a; a *= a; k >>= 1; } return ret; } template<typename T> pair<T,bool> Kthpower2(T a,T k){ //return {a^k,オーバーした?} オーバーフローは考慮する. T ret = 1,maxv = numeric_limits<T>::max(); while(k){ if(k&1){ if(isABmoreC(ret,a,maxv)) return {-1,true}; ret *= a; } if(k == 1) break; if(isABmoreC(a,a,maxv)) return {-1,true}; a *= a; k >>= 1; } return {ret,false}; } template<typename T> T Kthroot(T a,T k){ //return floor(a^(1/k)); assert(k > 0 && a >= 0); if(k == 1 || a <= 1) return a; T ret = pow(a,1.0/k); while(true){ auto [check,over] = Kthpower2(ret+1,k); if(over || check > a) break; ret++; } while(true){ auto [check,over] = Kthpower2(ret,k); if(!over && check <= a) break; ret--; } return ret; } template<typename T> T powmod(T a,T b,T m){//a^b(mod m)を返す. assert(b >= 0); __int128_t ret = 1,p = a; while(b){ if(b&1) ret = ret*p%m; p = p*p%m; b >>= 1; } return T(ret); } template<typename T> T divmod(T a,T b,T m){//a/b(mod m)を返す 素数mod必須. return (T)((__int128_t)a*powmod(b,m-2,m)%m); } 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); 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()){cout << "No\n"; return 0;} auto C2 = cycle.at(i).back(); cycle.at(i).pop_back(); for(int k=0; k<i; k++) Q.at(C1.at(k)) = C2.at(k),Q.at(C2.at(k)) = C1.at(k); } } else{ auto [ign,inc] = invgcd(2,i); inc = (i-inc)%i; for(auto &C : cycle.at(i)) for(int k=0; k<i; k++) Q.at(C.at((k+i-1%i)%i)) = C.at((k+inc)%i); } } for(auto &q : Q) q = (q-1+N)%N; cout << "Yes\n"; for(int i=0; i<N; i++) cout << Q.at(i)+1 << (i==N-1?"\n":" "); }