#990744 (C90) No.2793 Mancala Master

• 問題: No.2793 Mancala Master
• ユーザー: tails
• 提出日時: 2024-06-25 11:08:12
• 言語: C90 (gcc 11.4.0)
• 結果: AC
• 実行時間: 49 ms / 2,000 ms
• コード長: 1,007 bytes
• コンパイル時間: 978 ms
• コンパイル使用メモリ: 25,856 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-25 11:08:14
• 合計ジャッジ時間: 2,331 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	4 ms; 5,376 KB
testcase_03	AC	18 ms; 5,376 KB
testcase_04	AC	7 ms; 5,376 KB
testcase_05	AC	8 ms; 5,376 KB
testcase_06	AC	22 ms; 5,376 KB
testcase_07	AC	9 ms; 5,376 KB
testcase_08	AC	14 ms; 5,376 KB
testcase_09	AC	15 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	38 ms; 5,376 KB
testcase_12	AC	6 ms; 5,376 KB
testcase_13	AC	43 ms; 5,376 KB
testcase_14	AC	18 ms; 5,376 KB
testcase_15	AC	14 ms; 5,376 KB
testcase_16	AC	22 ms; 5,376 KB
testcase_17	AC	22 ms; 5,376 KB
testcase_18	AC	49 ms; 5,376 KB

コンパイルメッセージ

main.c: In function ‘main’:
main.c:54:9: warning: implicit declaration of function ‘write’ [-Wimplicit-function-declaration]
   54 |         write(1,wbuf,wp-wbuf);
      |         ^~~~~
main.c:55:9: warning: implicit declaration of function ‘_exit’ [-Wimplicit-function-declaration]
   55 |         _exit(0);
      |         ^~~~~
main.c:55: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() ({int _v=0,_c;while(_c=*rp++-48,_c>=0)_v=_v*10+_c;_v;})
#define WTHI(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 wt(v) if(v>=100000000){WTHI(v/10);*wp++=v%10+'0';}else{WTHI(v);}
#define rep(v,e) for(typeof(e)v=0;v<e;++v)
#define times(e) for(typeof(e)_=e;_--;)
#define MD 998244353

typedef unsigned long ulong;

char wbuf[1<<25];

long modpow(long a, ulong b){
	long r=1;
	while(b){
		if(b&1){
			r=r*a%MD;
		}
		b>>=1;
		a=a*a%MD;
	}
	return r;
}

int main(){
	char*wp=wbuf;
	rd_init();
	int t=rd();
	times(t){
		int n=rd();
		int k=rd();
		ulong z=1;
		ulong y=1;
		rep(i,n){
			y=y*k%MD;
			z=z*(y-1)%MD;
			ulong a=rd();
			ulong b=(i+1)*(a-1);
			ulong c=k;
			while(b){
				if(b&1){
					z=z*c%MD;
				}
				b>>=1;
				c=c*c%MD;
			}
		}
		wt(z);
		*wp++='\n';
	}
	write(1,wbuf,wp-wbuf);
	_exit(0);
}