結果

問題 No.2636 No Waiting in Vain
ユーザー tails
提出日時 2024-02-18 15:27:09
言語 C90
(gcc 12.3.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,165 bytes
コンパイル時間 854 ms
コンパイル使用メモリ 25,472 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-29 00:36:45
合計ジャッジ時間 2,053 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 44
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function ‘main’:
main.c:8:59: warning: implicit declaration of function ‘write’ [-Wimplicit-function-declaration]
    8 | #define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
      |                                                           ^~~~~
main.c:69:17: note: in expansion of macro ‘wt1’
   69 |                 wt1(z);
      |                 ^~~
main.c:71:9: warning: implicit declaration of function ‘_exit’ [-Wimplicit-function-declaration]
   71 |         _exit(0);
      |         ^~~~~
main.c:71:9: warning: incompatible implicit declaration of built-in function ‘_exit’ [-Wbuiltin-declaration-mismatch]

ソースコード

diff # 

#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_skip() while(*rp++>=48)
#define rd() ({long _v=0,_c;while(_c=*rp++-48,_c>=0)_v=_v*10+_c;_v;})
#define wt(v) ({ulong _z=v;do*--wp=_z%10+48;while(_z/=10);})
#define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
#define rrep3(v,s,e) for(typeof(e) v=e;--v>=s;)
#define min(a,b) (a<=b?a:b)
#define MD 998244353

typedef unsigned long ulong;

//--------
// ifac table

#define M 300001
#define ifac_Mm1 988693096
unsigned ifac[M];

void mkfac(){
	ulong x=ifac_Mm1;
	for(long i=M;i--;){
		ifac[i]=x;
		x=x*i%MD;
	}
}

//--------

int inverse(int a){
	int b=MD;
	int u=1;
	int v=0;
	int s,t;
	while(b){
		t=a/b;
		s=b; b=a-t*b; a=s;
		s=v; v=u-t*v; u=s;
	}
	if(u<0){
		u+=MD;
	}
	return u;
}

int main(){
	int k,d;
	{
		rd_init();
		rd_skip();
		k=rd();
		d=0;
		for(int a,b=0;a=*rp++,a!=10;b=a){
			if(a=='N'){
				k-=1;
				d+=b!='C';
			}
		}
	}
	mkfac();
	{
		ulong z=0;
		rrep3(i,1,min(-k,d)+1){
			z=(z+(ulong)ifac[i]*ifac[d-i])%MD;
		}
		z=z*inverse(ifac[d])%MD;
		wt1(z);
	}
	_exit(0);
}
0