#952206 (C90) No.2636 No Waiting in Vain

• 問題: No.2636 No Waiting in Vain
• ユーザー: tails
• 提出日時: 2024-02-18 15:27:09
• 言語: C90 (gcc 11.4.0)
• 結果: AC
• 実行時間: 5 ms / 2,000 ms
• コード長: 1,165 bytes
• コンパイル時間: 681 ms
• コンパイル使用メモリ: 25,472 KB
• 実行使用メモリ: 6,676 KB
• 最終ジャッジ日時: 2024-02-18 15:27:12
• 合計ジャッジ時間: 2,257 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 6,676 KB
testcase_01	AC	2 ms; 6,676 KB
testcase_02	AC	2 ms; 6,676 KB
testcase_03	AC	2 ms; 6,676 KB
testcase_04	AC	5 ms; 6,676 KB
testcase_05	AC	5 ms; 6,676 KB
testcase_06	AC	4 ms; 6,676 KB
testcase_07	AC	5 ms; 6,676 KB
testcase_08	AC	4 ms; 6,676 KB
testcase_09	AC	4 ms; 6,676 KB
testcase_10	AC	4 ms; 6,676 KB
testcase_11	AC	4 ms; 6,676 KB
testcase_12	AC	4 ms; 6,676 KB
testcase_13	AC	4 ms; 6,676 KB
testcase_14	AC	4 ms; 6,676 KB
testcase_15	AC	5 ms; 6,676 KB
testcase_16	AC	4 ms; 6,676 KB
testcase_17	AC	4 ms; 6,676 KB
testcase_18	AC	5 ms; 6,676 KB
testcase_19	AC	5 ms; 6,676 KB
testcase_20	AC	4 ms; 6,676 KB
testcase_21	AC	4 ms; 6,676 KB
testcase_22	AC	4 ms; 6,676 KB
testcase_23	AC	5 ms; 6,676 KB
testcase_24	AC	2 ms; 6,676 KB
testcase_25	AC	3 ms; 6,676 KB
testcase_26	AC	3 ms; 6,676 KB
testcase_27	AC	3 ms; 6,676 KB
testcase_28	AC	3 ms; 6,676 KB
testcase_29	AC	3 ms; 6,676 KB
testcase_30	AC	2 ms; 6,676 KB
testcase_31	AC	3 ms; 6,676 KB
testcase_32	AC	3 ms; 6,676 KB
testcase_33	AC	2 ms; 6,676 KB
testcase_34	AC	2 ms; 6,676 KB
testcase_35	AC	3 ms; 6,676 KB
testcase_36	AC	2 ms; 6,676 KB
testcase_37	AC	3 ms; 6,676 KB
testcase_38	AC	2 ms; 6,676 KB
testcase_39	AC	2 ms; 6,676 KB
testcase_40	AC	3 ms; 6,676 KB
testcase_41	AC	2 ms; 6,676 KB
testcase_42	AC	3 ms; 6,676 KB
testcase_43	AC	2 ms; 6,676 KB

コンパイルメッセージ

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]

ソースコード

#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);
}