結果

問題 No.3256 Permutation Equation
ユーザー Taiki0715
提出日時 2025-09-06 15:47:25
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,467 bytes
コンパイル時間 3,647 ms
コンパイル使用メモリ 292,300 KB
実行使用メモリ 12,616 KB
最終ジャッジ日時 2025-09-06 15:47:46
合計ジャッジ時間 14,302 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T1,typename T2,typename T3>istream &operator>>(istream &is,tuple<T1,T2,T3>&a){is>>std::get<0>(a)>>std::get<1>(a)>>std::get<2>(a);return is;}
template<typename T,size_t n>istream &operator>>(istream &is,array<T,n>&a){for(auto&i:a)is>>i;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define overload3(_1,_2,_3,name,...) name
#define rep1(i,n) for(int i=0;i<(int)(n);i++)
#define rep2(i,l,r) for(int i=(int)(l);i<(int)(r);i++)
#define rep(...) overload3(__VA_ARGS__,rep2,rep1)(__VA_ARGS__)
#define reps(i,l,r) rep2(i,l,r)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcountll(x)
#define fin(x) return cout<<(x)<<'\n',static_cast<void>(0)
#define yn(x) cout<<((x)?"Yes\n":"No\n")
#define uniq(x) sort(all(x)),x.erase(unique(all(x)),x.end())
template<typename T>
inline int fkey(vector<T>&z,T key){return lower_bound(z.begin(),z.end(),key)-z.begin();}
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
#ifdef LOCAL
#include<debug.h>
#define SWITCH(a,b) (a)
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
#define SWITCH(a,b) (b)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
struct Timer{
  clock_t start;
  Timer(){
    start=clock();
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout<<fixed<<setprecision(16);
  }
  inline double now(){return (double)(clock()-start)/1000;}
  #ifdef LOCAL
  ~Timer(){
    cerr<<"time:";
    cerr<<now();
    cerr<<"ms\n";
  }
  #endif
}timer;
void SOLVE();
int main(){
  int testcase=1;
  //cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
}
struct UnionFind{
private:
  std::vector<int>par;
  int cs;
public:
  UnionFind(int n):par(n,-1),cs(n){}
  UnionFind(){}
  int root(int u){
    if(par[u]<0)return u;
    return par[u]=root(par[u]);
  }
  bool merge(int u,int v){
    int ru=root(u),rv=root(v);
    if(ru==rv)return false;
    if(par[ru]<par[rv])std::swap(ru,rv);
    par[rv]+=par[ru];
    par[ru]=rv;
    cs--;
    return true;
  }
  bool same(int u,int v){return root(u)==root(v);}
  int size(int u=-1){return u==-1?cs:-par[root(u)];}
  void reset(int n){
    cs=n;
    std::fill(par.begin(),par.begin()+n,-1);
  }
  std::vector<std::vector<int>>get_all(){
    std::vector<std::vector<int>>res(par.size());
    for(int i=0;i<(int)par.size();i++)res[root(i)].push_back(i);
    int p=0;
    for(int i=0;i<(int)res.size();i++){
      if(!res[i].empty()){
        if(i==p)p++;
        else res[p++]=std::move(res[i]);
      }
    }
    res.resize(p);
    return res;
  }
};
void SOLVE(){
  int n;
  cin>>n;
  vector<int>p(n);
  cin>>p;
  rep(i,n)if(p[i]==n)p[i]=0;
  UnionFind uf(n);
  rep(i,n)uf.merge(i,p[i]);
  auto g=uf.get_all();
  vector<int>ans(n);
  vector<vector<int>>even;
  for(auto vs:g){
    if(ssize(vs)&1){
      int m=ssize(vs);
      int s=vs[0],t=vs[0];
      rep(i,(m+1)/2)t=p[t];
      rep(i,m){
        ans[s]=t;
        s=p[s],t=p[t];
      }
    }
    else even.push_back(vs);
  }
  sort(all(even),[](auto&lhs,auto&rhs){return ssize(lhs)<ssize(rhs);});
  if(even.size()%2)fin("No");
  for(int i=0;i<ssize(even);i+=2){
    if(ssize(even[i])!=ssize(even[i+1]))fin("No");
    rep(j,ssize(even[i])){
      ans[even[i][j]]=even[i+1][j];
      ans[even[i+1][j]]=even[i][(j+1)%ssize(even[i])];
    }
  }
  cout<<"Yes\n";
  rep(i,n)if(--ans[i]<0)ans[i]+=n;
  rep(i,n)cout<<ans[i]+1<<" \n"[i+1==n];
}
0