結果
| 問題 | No.1897 Sum of 2nd Max |
| コンテスト | |
| ユーザー |
lam6er
|
| 提出日時 | 2025-03-20 20:20:03 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 239 ms / 2,000 ms |
| コード長 | 621 bytes |
| 記録 | |
| コンパイル時間 | 141 ms |
| コンパイル使用メモリ | 82,244 KB |
| 実行使用メモリ | 62,804 KB |
| 最終ジャッジ日時 | 2025-03-20 20:21:03 |
| 合計ジャッジ時間 | 5,000 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 30 |
ソースコード
MOD = 998244353
N, K = map(int, input().split())
result = 0
for x in range(1, K + 1):
a = pow(x, N-1, MOD)
b = pow(x-1, N-1, MOD)
c = a * x % MOD
d = b * (x-1) % MOD
# Compute term1: (K - x) * N * (a - b)
term1_part = ((K - x) % MOD) * (N % MOD) % MOD
diff_ab = (a - b) % MOD
term1 = term1_part * diff_ab % MOD
# Compute term2: (c - d) - N * b
term2_part = (c - d) % MOD
term2_part = (term2_part - (N % MOD) * b) % MOD
term2 = term2_part
total = (term1 + term2) % MOD
total = total * x % MOD
result = (result + total) % MOD
print(result)
lam6er