結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー tails
提出日時 2021-06-25 19:49:43
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 12 ms / 2,500 ms
コード長 1,210 bytes
コンパイル時間 240 ms
コンパイル使用メモリ 32,000 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-25 07:45:13
合計ジャッジ時間 1,045 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:48:1: warning: return type defaults to 'int' [-Wimplicit-int]
   48 | main(){
      | ^~~~
main.c: In function 'main':
main.c:52:9: warning: implicit declaration of function 'read'; did you mean 'rd'? [-Wimplicit-function-declaration]
   52 |         read(0,rwbuf,sizeof rwbuf);
      |         ^~~~
      |         rd
main.c:68:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   68 |         write(1,wp,rwbuf+sizeof rwbuf-wp);
      |         ^~~~~
main.c:69:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   69 |         _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
unsigned sieve[SIEVE_N/2+1][2];
void mksieve(){
	for(long i=3;i*i<SIEVE_N;i+=2){
		if(sieve[i>>1][0]==0){
			long k=i;
			for(long j=i*i>>1;j<SIEVE_N/2;j+=i){
				sieve[j][0]=i;
				sieve[j][1]=k;
				k+=2;
			}
		}
	}
}

#define MD 998244353

unsigned pwmemo[SIEVE_N];

long pw(long a,long n){
	unsigned long r;
	if(!(a&1)&&a!=2){
		r=(unsigned long)pwmemo[2]*(unsigned long)pwmemo[a>>1]%MD;
	}else if(sieve[a>>1][0]){
		r=(unsigned long)pwmemo[sieve[a>>1][0]]*(unsigned long)pwmemo[sieve[a>>1][1]]%MD;
	}else{
		r=1;
		unsigned 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