結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー tpc011854
提出日時 2021-06-18 23:20:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 595 ms / 2,500 ms
コード長 1,123 bytes
コンパイル時間 257 ms
コンパイル使用メモリ 32,896 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-22 21:28:08
合計ジャッジ時間 7,973 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
const long long mod = 998244353;

long long power(long long a,long long b){
	long long res = 1;
	while(b){
		if(b&1) res = res * a%mod;
		a = a*a%mod;
		b /= 2;
	}
	return res;
}

long long NotBigger(long long k,long long n){
	long long digit = (1+k)*k/2%mod;
	n = n%mod;
	return digit*(n%mod)%mod;
}

long long inRange(long long l,long long r,long long n){
	if(l>r) return 0;
	long long digit = (l+r)*(r-l+1)/2%mod;
	long long gap = r-l+1;
	long long res =  digit*power(gap,n-1)%mod;
	res = res*(n%mod)%mod;
	return res;
}

int main(){
	long long n,m;
	scanf("%lld%lld",&n,&m);
	long long pos = 0, neg = 0;
	for(int i = 1; i <= m; i++){
		long long way = inRange(1,i,n);
		way = way - inRange(1,i-1,n);
		if(way < 0) way += mod;
		//printf("pway = %lld\n",way);
		pos = (pos + way * (long long)i%mod)%mod;
		way = inRange(i,m,n);
		way = way - inRange(i+1,m,n);
		if(way < 0) way += mod;
		//printf("nway = %lld\n",way);
		neg = (neg + way * (long long)i%mod)%mod;
	}
	//printf("pos = %lld, neg = %lld\n",pos,neg);
	long long ans = pos - neg;
	if(ans < 0) ans += mod;
	printf("%lld\n",ans);
	return 0;
}
0