結果

問題 No.1552 Simple Dice Game
ユーザー tailstails
提出日時 2021-06-25 19:13:34
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 19 ms / 2,500 ms
コード長 988 bytes
コンパイル時間 155 ms
コンパイル使用メモリ 31,360 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-25 07:45:04
合計ジャッジ時間 1,164 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:26:1: warning: return type defaults to 'int' [-Wimplicit-int]
   26 | main(){
      | ^~~~
main.c: In function 'main':
main.c:30:9: warning: implicit declaration of function 'read'; did you mean 'rd'? [-Wimplicit-function-declaration]
   30 |         read(0,rwbuf,sizeof rwbuf);
      |         ^~~~
      |         rd
main.c:60:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   60 |         write(1,wp,rwbuf+sizeof rwbuf-wp);
      |         ^~~~~
main.c:61:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   61 |         _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];

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,r=0;
	rep(d,m+1){
		a+=r;
		if(sieve[d]){
			r=(long)pwmemo[sieve[d]]*(long)pwmemo[d/sieve[d]]%MD;
		}else{
			r=1;
			long n=e,b=d;
			while(n){
				if(n&1){
					r=r*b%MD;
				}
				n>>=1;
				b=b*b%MD;
			}
		}
		pwmemo[d]=r;
	}
	a=r*(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