結果

問題 No.3441 Sort Permutation 2
コンテスト
ユーザー 👑 tails
提出日時 2026-02-06 23:13:29
言語 C
(gcc 15.2.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 1,356 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 327 ms
コンパイル使用メモリ 40,256 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-02-06 23:13:33
合計ジャッジ時間 4,227 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#pragma GCC diagnostic ignored "-Wimplicit-function-declaration"
#pragma GCC diagnostic ignored "-Wbuiltin-declaration-mismatch"
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

#define rd_init() char*rp=({char*mmap(...);mmap(0l,1l<<25,1,2,0,0ll);})
#define rd() ({int _v=0,_c;while(_c=*rp++-48,_c>=0)_v=_v*10+_c;_v;})
#define wt(v) ({unsigned _z=v,_n=0;long _d=0;while(++_n,_d=_d<<8|0x30|_z%10,_z/=10);*(long*)wp=_d;wp+=_n;})
#define rep(v,e) for(typeof(e)v=0;v<e;++v)
#define reps(v,s,e) for(typeof(e)v=s;v<e;++v)
#define abs(a) ((a)<0?-(a):(a))

int gcd(int a,int b){
	if(a==0){
		return b;
	}
	int x=__builtin_ctz(a);
	int y=__builtin_ctz(b);
	a>>=x;
	b>>=y;
	int c;
	while(c=a-b){
		c>>=__builtin_ctz(c);
		if(c<0){
			b=-c;
		}else{
			a=c;
		}
	}
	return a<<(x<=y?x:y);
}

char wbuf[1<<25];
int p[200000];
int c[200000];
char v[200000];
int n;

void f1(){
	rd_init();
	n=rd();
	rep(i,n){
		p[i]=rd()-1;
	}
}

void f2(){
	rep(i,n){
		if(!v[i]){
			int x=i,g=0,d=0;
			do{
				g=gcd(g,abs(p[x]-x));
				++d;
				v[x]=1;
				x=p[x];
			}while(x!=i);
			c[g]+=d-1;
		}
	}
}

void f3(){
	reps(i,2,n){
		if(v[i]){
			for(int j=(n-1)/i,k=j*i;j;j-=1,k-=i){
				c[j]+=c[k];
				v[k]=0;
			}
		}
	}
}

void f4(){
	char*wp=wbuf;
	reps(i,1,n){
		wt(c[i]);
		*wp++='\n';
	}
	write(1,wbuf,wp-wbuf);
}

int main(){
	f1();
	f2();
	f3();
	f4();
	_exit(0);
}
0