#990745 (C90) No.2793 Mancala Master

• 問題: No.2793 Mancala Master
• ユーザー: tails
• 提出日時: 2024-06-25 11:13:33
• 言語: C90 (gcc 11.4.0)
• 結果: AC
• 実行時間: 24 ms / 2,000 ms
• コード長: 1,077 bytes
• コンパイル時間: 516 ms
• コンパイル使用メモリ: 26,240 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-25 11:13:35
• 合計ジャッジ時間: 2,003 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	0 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	1 ms; 5,376 KB
testcase_16	AC	23 ms; 5,376 KB
testcase_17	AC	24 ms; 5,376 KB
testcase_18	AC	3 ms; 5,376 KB

コンパイルメッセージ

main.c: In function ‘main’:
main.c:58:9: warning: implicit declaration of function ‘write’ [-Wimplicit-function-declaration]
   58 |         write(1,wbuf,wp-wbuf);
      |         ^~~~~
main.c:59:9: warning: implicit declaration of function ‘_exit’ [-Wimplicit-function-declaration]
   59 |         _exit(0);
      |         ^~~~~
main.c:59: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 b=0;
		rep(i,n){
			ulong a=rd();
			b+=(i+1)*(a-1);
			b-=b>=(ulong)(MD-1)*(MD-1)?(ulong)(MD-1)*(MD-1):0;
		}
		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);
}