結果

問題 No.1897 Sum of 2nd Max
ユーザー publfl
提出日時 2022-04-08 22:45:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 65 ms / 2,000 ms
コード長 861 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 33,792 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-28 13:35:22
合計ジャッジ時間 2,191 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#define MOD 998244353

long long int power(long long int a, long long int b)
{
	long long int ans = 1;
	long long int k = a;
	while(b)
	{
		if(b%2==1) ans*=k, ans %=MOD;
		b/=2;
		k*=k, k%=MOD;
	}
	return ans;
}

long long int count[200010];
int main()
{
	int a,b;
	scanf("%d%d",&a,&b);
	
	long long int ans = 0;
	for(int i=1;i<=b;i++)
	{
		long long int val = power(i,a);
		val -= a*power(i-1,a-1), val %= MOD, val += MOD, val %= MOD;
		val -= power(i-1,a), val %= MOD, val += MOD, val %= MOD;
		val *= i, val %= MOD;
		ans += val, ans %= MOD;
	}
	
	for(int i=1;i<=b;i++) count[i] = power(i,a-1);
	for(int i=b;i>=1;i--) count[i] = (count[i]-count[i-1]+MOD)%MOD;
	
	long long int S = 0;
	for(int i=1;i<=b;i++)
	{
		S += (i-1)*count[i-1];
		S %= MOD;
		long long int val = S*a;
		val %= MOD;
		ans += val, ans %= MOD;
	}
	printf("%lld",ans);
}
0