結果

問題 No.1897 Sum of 2nd Max
コンテスト
ユーザー ygussany
提出日時 2022-03-20 10:43:48
言語 C(gnu17)
(gcc 15.2.0)
コンパイル:
gcc-15 -O2 -std=gnu17 -Wno-error=implicit-function-declaration -Wno-error=implicit-int -Wno-error=incompatible-pointer-types -Wno-error=int-conversion -DONLINE_JUDGE -o a.out _filename_ -lm
実行:
./a.out
結果
AC  
実行時間 75 ms / 2,000 ms
コード長 632 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 213 ms
コンパイル使用メモリ 40,472 KB
最終ジャッジ日時 2026-02-22 08:58:37
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include <stdio.h>

const int Mod = 998244353;

long long pow_mod(int n, long long k)
{
	long long N, ans = 1;
	for (N = n; k > 0; k >>= 1, N = N * N % Mod) if (k & 1) ans = ans * N % Mod;
	return ans;
}

int main()
{
	int N, K;
	scanf("%d %d", &N, &K);
	
	int i;
	long long ans = 1 + (long long)N * (K - 1) % Mod, tmp;
	for (i = 2; i <= K; i++) {
		tmp = 0;
		if (i < K) tmp += (pow_mod(i, N - 1) - pow_mod(i - 1, N - 1) + Mod) * (K - i) % Mod * N % Mod;
		tmp += pow_mod(i, N) - pow_mod(i - 1, N) + Mod - pow_mod(i - 1, N - 1) * N % Mod + Mod;
		ans += tmp * i % Mod;
	}
	printf("%lld\n", ans % Mod);
	fflush(stdout);
	return 0;
}
0