AtCoder Beginner Contest #013 D問題 - 阿弥陀

Source

AtCoder Beginner Contest #013
問題文

問題概要

縦棒が $N$ 本あるあみだくじを考える．
横棒の最初の $M$ 本に関しては，$k$ 番目の横棒は $A_k$ 番目の縦棒と $A_{k}+1$ 番目の縦棒を結んでいる．
そのような $M$ 本の横棒の組が $D$ 回繰り返される．
上端のどの棒が下端のどの棒に対応するかをすべて求める問題．

解法

まず，最初の $M$ 本を愚直に順番にスワップさせて，最初の $M$ 本でどのように置換されるかを求める．
その置換を巡回置換の積に分解して，その巡回置換において，$D$ 個分循環させれば良い．
時間計算量 $O(N+M)$．

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

#define ll long long

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void reader(int *x, int *y, int *z){reader(x);reader(y);reader(z);}
void reader(ll *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}return s;}

void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}
void writer(ll x, char c){int i,sz=0,m=0;char buf[20];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}
void writer(const char c[]){int i;for(i=0;c[i]!='\0';i++)mypc(c[i]);}

int N, M, D;
int p[110000];


int vis[110000];
int res[110000];
int arr[110000], sz;

int main(){
  int i, j, k;

  reader(&N,&M,&D);
  rep(i,N) p[i] = i;
  while(M--){
    reader(&i);
    swap(p[i-1], p[i]);
  }

  rep(i,N) if(!vis[i]){
    sz = 0;
    k = i;
    while(!vis[k]){
      arr[sz++] = k;
      vis[k] = 1;
      k = p[k];
    }
    rep(k,sz) res[arr[k]] = arr[(k+D)%sz];
  }

  rep(i,N) arr[i] = res[i];
  rep(i,N) res[arr[i]] = i;
  rep(i,N) writer(res[i]+1,'\n');

  return 0;
}

Current time: 2024年04月26日06時36分39秒
Last modified: 2014年11月18日03時45分03秒 (by laycrs)
Tags: Competitive_Programming AtCoder AtCoder_Beginner_Contest ABC013 ABC_D
トップページに戻る

Logged in as: unknown user (not login)

ログイン：