結果

問題 No.3256 Permutation Equation
ユーザー Yoyoyo8128
提出日時 2025-09-05 22:44:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 7,945 bytes
コンパイル時間 5,072 ms
コンパイル使用メモリ 271,368 KB
実行使用メモリ 10,432 KB
最終ジャッジ日時 2025-09-05 22:45:10
合計ジャッジ時間 11,217 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma region Yoyoyo


#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using ld = long double;
using i128 = __int128_t;
using pii = pair<int, int>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
typedef vector<ll> vi;
typedef vector<vi> vvi;
const string Yes = "Yes";
const string No = "No";
const string YES = "YES";
const string NO = "NO";
const string abc = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const ll MOD = 1000000007;
const ll mod = 998244353;
const long long inf = 1ll << 60;
const long double PI = 3.1415926535897932384626;
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define all(x) (x).begin(), (x).end()
#define fi first
#define se second
#define faster ios::sync_with_stdio(false);cin.tie(nullptr);
const vector<int> dx = {0, 1, 0, -1, 1, -1, -1, 1};
const vector<int> dy = {1, 0, -1, 0, 1, 1, -1, -1};



#define YESNO(T)               \
    if (T)                     \
    {                          \
        cout << "YES" << endl; \
    }                          \
    else                       \
    {                          \
        cout << "NO" << endl;  \
    }
#define yesno(T)               \
    if (T)                     \
    {                          \
        cout << "yes" << endl; \
    }                          \
    else                       \
    {                          \
        cout << "no" << endl;  \
    }
#define YesNo(T)               \
    if (T)                     \
    {                          \
        cout << "Yes" << endl; \
    }                          \
    else                       \
    {                          \
        cout << "No" << endl;  \
    }
#define inV(vec)                        \
    for (ll i = 0; i < vec.size(); i++) \
        cin >> vec[i];
#define outV(vec)                       \
    for (ll i = 0; i < vec.size(); i++) \
    {                                   \
        cout << vec[i] << " ";          \
    }                                   \
    cout << endl;
#define print(s) cout << s << endl;
#define updiv(n, x) (n + x - 1) / x
#define rounddiv(n, x) (ll)((double)(n) / (double)(x) + 0.5)
#define fix(n)                \
    fixed << setprecision(n)
template <typename T>
inline bool chmax(T &a, T b) { return ((a < b) ? (a = b, true) : (false)); }
template <typename T>
inline bool chmin(T &a, T b) { return ((a > b) ? (a = b, true) : (false)); }
template <class T>
ll sum(const T &a) { return accumulate(all(a), 0LL); }

// pair_out
template <typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &p)
{
    os << p.first << " " << p.second;
    return os;
}
// pair_in
template <typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &p)
{
    is >> p.first >> p.second;
    return is;
}
// vector_out
template <typename T>
ostream &operator<<(ostream &os, const vector<T> &v)
{
    int s = (int)v.size();
    for (int i = 0; i < s; i++)
        os << (i ? " " : "") << v[i];
    return os;
}
// vector_in
template <typename T>
istream &operator>>(istream &is, vector<T> &v)
{
    for (auto &x : v)
        is >> x;
    return is;
}
//__int128_t_in
istream &operator>>(istream &is, __int128_t &x)
{
    string S;
    is >> S;
    x = 0;
    int flag = 0;
    for (auto &c : S)
    {
        if (c == '-')
        {
            flag = true;
            continue;
        }
        x *= 10;
        x += c - '0';
    }
    if (flag)
        x = -x;
    return is;
}
//__uint128_t_in
istream &operator>>(istream &is, __uint128_t &x)
{
    string S;
    is >> S;
    x = 0;
    for (auto &c : S)
    {
        x *= 10;
        x += c - '0';
    }
    return is;
}
//__int128_t_out
ostream &operator<<(ostream &os, __int128_t x)
{
    if (x == 0)
        return os << 0;
    if (x < 0)
        os << '-', x = -x;
    string S;
    while (x)
        S.push_back('0' + x % 10), x /= 10;
    reverse(begin(S), end(S));
    return os << S;
}
//__uint128_t_out
ostream &operator<<(ostream &os, __uint128_t x)
{
    if (x == 0)
        return os << 0;
    string S;
    while (x)
        S.push_back('0' + x % 10), x /= 10;
    reverse(begin(S), end(S));
    return os << S;
}
// vector<vector>_out
template <typename T>
ostream &operator<<(ostream &os, const vector<vector<T>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << endl;
    }
    return os;
}
// vector<vector<vector>>_out
template <typename T>
ostream &operator<<(ostream &os, const vector<vector<vector<T>>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << "i = " << i << endl;
        os << v[i];
    }
    return os;
}
// map_out
template <typename T, typename S>
ostream &operator<<(ostream &os, const map<T, S> &m)
{
    for (auto &[key, val] : m)
    {
        os << key << ":" << val << " ";
    }
    return os;
}
// set_out
template <typename T>
ostream &operator<<(ostream &os, const set<T> &st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}
// multiset_out
template <typename T>
ostream &operator<<(ostream &os, const multiset<T> &st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}
// queue_out
template <typename T>
ostream &operator<<(ostream &os, queue<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop();
    }
    return os;
}
// deque_out
template <typename T>
ostream &operator<<(ostream &os, deque<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop_front();
    }
    return os;
}
// stack_out
template <typename T>
ostream &operator<<(ostream &os, stack<T> st)
{
    while (st.size())
    {
        os << st.top() << " ";
        st.pop();
    }
    return os;
}
// priority_queue_out
template <class T, class Container, class Compare>
ostream &operator<<(ostream &os, priority_queue<T, Container, Compare> pq)
{
    while (pq.size())
    {
        os << pq.top() << " ";
        pq.pop();
    }
    return os;
}



template<class... Args>
void debug_out(Args... args) {
    int _i = 0;
    ((cerr << ( _i++ ? ", " : " ") << args), ...);
    cerr << "\n";
}

#pragma endregion Yoyoyo




#define debug(...)

int main(){
    faster;
    vector<vector<ll>>loop(0);
    ll N;
    cin>>N;
    vector<ll>A(N);
    cin>>A;
    for(int i=0;i<N;i++){
        A[i]%=N;
    }
    vector<bool>vis(N,0);
    for(int i=0;i<N;i++){
        if(!vis[i]){
            vector<ll>tmp(0);
            
            ll now=i;
            while(now!=i || (!vis[now])){
                tmp.pb(now);
                vis[now]=true;
                now=A[now];
            }
            loop.pb(tmp);
            debug(tmp);
        }
    }
    vector<ll>ans(N);
    sort(all(loop),[](vector<ll>& a,vector<ll>& b){return a.size()<b.size();});
    debug(loop);
    vector<ll>siz(N+1);
    for(int i=0;i<loop.size();i++){
        siz[loop[i].size()]++;
    }
    debug(siz);
    for(int i=0;i<=N;i++){
        if(i%2==0 && siz[i]%2==1){
            print(No);exit(0);
        }
    }
    for(int i=0;i<loop.size();i++){
        if(loop[i].size()%2==1){
            for(int j=0;j<loop[i].size();j++){
                ans[loop[i][j]]=loop[i][((j+(loop[i].size()+1)/2))%(loop[i].size())];
            }
        }else{
            for(int j=0;j<loop[i].size();j++){
                ans[loop[i][j]]=loop[i+1][j];
            }
            for(int j=0;j<loop[i].size();j++){
                ans[loop[i+1][j]]=loop[i][(j+1)%(loop[i].size())];
            }
            i++;
        }
    }
    for(int i=0;i<N;i++){
        if(ans[i]==0){
            ans[i]+=N;
        }
    }
    print(Yes);
    print(ans);
}
0