結果

問題 No.3256 Permutation Equation
ユーザー GOTKAKO
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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":" ");
} 
0