結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー tails
提出日時 2021-06-25 18:57:13
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 20 ms / 2,500 ms
コード長 1,024 bytes
コンパイル時間 410 ms
コンパイル使用メモリ 31,232 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-25 07:45:02
合計ジャッジ時間 1,327 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:45:1: warning: return type defaults to 'int' [-Wimplicit-int]
   45 | main(){
      | ^~~~
main.c: In function 'main':
main.c:49:9: warning: implicit declaration of function 'read'; did you mean 'rd'? [-Wimplicit-function-declaration]
   49 |         read(0,rwbuf,sizeof rwbuf);
      |         ^~~~
      |         rd
main.c:65:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   65 |         write(1,wp,rwbuf+sizeof rwbuf-wp);
      |         ^~~~~
main.c:66:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   66 |         _exit(0);
      |         ^~~~~
      |         _Exit

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

#define rd(v) long v=0;{int _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}
#define wt(v) {long _z=v;do*--wp=_z%10+48;while(_z/=10);}
#define rep(v,e) for(long v=0;v<e;++v)

#define SIEVE_N 500002
int sieve[SIEVE_N];
void mksieve(){
	for(long i=2;i<SIEVE_N;++i){
		if(sieve[i]==0){
			for(long j=i*i;j<SIEVE_N;j+=i){
				if(sieve[j]==0){
					sieve[j]=i;
				}
			}
		}
	}
}

#define MD 998244353

int pwmemo[SIEVE_N];

long pw(long a,long n){
	long r;
	if(sieve[a]){
		r=(long)pwmemo[sieve[a]]*(long)pwmemo[a/sieve[a]]%MD;
	}else{
		r=1;
		long b=a;
		while(n){
			if(n&1){
				r=r*b%MD;
			}
			n>>=1;
			b=b*b%MD;
		}
	}
	pwmemo[a]=r;
	return r;
}

main(){
	mksieve();

	char rwbuf[64];
	read(0,rwbuf,sizeof rwbuf);
	char*rp=rwbuf;
	rd(n);
	rd(m);

	long e=n%(MD-1);
	long a=0;
	rep(d,m){
		a=a+pw(d,e);
	}
	a=pw(m,e)*(m-1)%MD*(MD+1>>1)%MD-a%MD;
	a=a*(n%MD)%MD*(m+1)%MD;
	a+=a<0?MD:0;

	char*wp=rwbuf+sizeof rwbuf;
	wt(a);
	write(1,wp,rwbuf+sizeof rwbuf-wp);
	_exit(0);
}
0