結果
問題 |
No.3256 Permutation Equation
|
ユーザー |
![]() |
提出日時 | 2025-09-05 23:39:53 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 143 ms / 2,000 ms |
コード長 | 2,969 bytes |
コンパイル時間 | 2,338 ms |
コンパイル使用メモリ | 207,404 KB |
実行使用メモリ | 77,272 KB |
最終ジャッジ日時 | 2025-09-05 23:40:05 |
合計ジャッジ時間 | 11,820 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 48 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef signed long long ll; #define _P(...) (void)printf(__VA_ARGS__) #define FOR(x,to) for(x=0;x<(to);x++) #define FORR(x,arr) for(auto& x:arr) #define FORR2(x,y,arr) for(auto& [x,y]:arr) #define ALL(a) (a.begin()),(a.end()) #define ZERO(a) memset(a,0,sizeof(a)) #define MINUS(a) memset(a,0xff,sizeof(a)) template<class T> bool chmax(T &a, const T &b) { if(a<b){a=b;return 1;}return 0;} template<class T> bool chmin(T &a, const T &b) { if(a>b){a=b;return 1;}return 0;} //------------------------------------------------------- int N,P[202020],S[202020],ret[202020]; template<int NV> struct FunctionGraph { //input int E[NV]; //output int N; int NG; //閉路数 int nex[61][NV]; // 2^i先の点 vector<int> RE[NV]; // 逆向き辺 vector<int> G[NV]; // 閉路 int GS[NV]; // 閉路長 int NC[NV]; // subtreeの要素数 int Gid[NV],Gindex[NV]; // 閉路IDとその中のindex。閉路外の場合、閉路IDは最寄りの閉路だがindexは-1 int GV[NV], D[NV]; //最寄りの閉路の点とそこまでの距離 int dfs(int cur,int id,int d) { GV[cur]=id; Gid[cur]=Gid[id]; D[cur]=d; NC[cur]=1; FORR(e,RE[cur]) if(Gindex[e]==-1) NC[cur]+=dfs(e,id,d+1); return NC[cur]; } int move(int cur,ll step) { // step先に進む int i; FOR(i,60) if(step&(1LL<<i)) cur=nex[i][cur]; return cur; } void build(int N_) { N=N_; int i,j; vector<int> in(N); FOR(i,N) { RE[i].clear(); G[i].clear(); Gindex[i]=-2; nex[0][i]=E[i]; } NG=0; FOR(i,N) { RE[E[i]].push_back(i); in[E[i]]++; } queue<int> Q; FOR(i,N) if(in[i]==0) Q.push(i); while(Q.size()) { int cur=Q.front(); Q.pop(); Gindex[cur]=-1; if(--in[E[cur]]==0) Q.push(E[cur]); } FOR(i,N) if(Gindex[i]==-2) { G[NG].push_back(i); while(1) { int x=G[NG].back(); Gid[x]=NG; Gindex[x]=G[NG].size()-1; x=E[x]; if(x==G[NG][0]) break; G[NG].push_back(x); } GS[NG]=G[NG].size(); NG++; } FOR(i,N) if(Gindex[i]>=0) dfs(i,i,0); FOR(j,60) FOR(i,N) nex[j+1][i]=nex[j][nex[j][i]]; } }; FunctionGraph<201010> fg; int cand[202020]; void solve() { int i,j,k,l,r,x,y; string s; cin>>N; FOR(i,N) { cin>>P[i]; S[i]=P[i]%N; fg.E[i]=S[i]; } MINUS(cand); MINUS(ret); fg.build(N); FOR(i,fg.NG) { auto G=fg.G[i]; x=fg.GS[i]; if(x%2==1) { FOR(j,x) ret[G[j]]=G[(j+(x+1)/2)%x]; } else if(cand[x]==-1) { cand[x]=i; } else { auto G2=fg.G[cand[x]]; cand[x]=-1; FOR(j,x) { ret[G[j]]=G2[j]; ret[G2[j]]=G[(j+1)%x]; } } } FOR(i,N) if(ret[i]==-1) { cout<<"No"<<endl; return; } cout<<"Yes"<<endl; FOR(i,N) { ret[i]=(ret[i]+N-1)%N; cout<<ret[i]+1<<" "; } cout<<endl; } int main(int argc,char** argv){ string s;int i; if(argc==1) ios::sync_with_stdio(false), cin.tie(0); FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin); cout.tie(0); solve(); return 0; }