結果
問題 | No.1897 Sum of 2nd Max |
ユーザー | 👑 rin204 |
提出日時 | 2022-04-08 22:08:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 262 ms / 2,000 ms |
コード長 | 305 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 82,028 KB |
実行使用メモリ | 73,412 KB |
最終ジャッジ日時 | 2024-11-28 12:47:54 |
合計ジャッジ時間 | 4,767 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
MOD = 998244353 n, k = map(int, input().split()) dp = [0] * k for i in range(k): dp[i] = pow(k - i, n, MOD) + pow(k - i, n - 1, MOD) * i * n % MOD dp[i] %= MOD for i in range(k - 1): dp[i] -= dp[i + 1] ans = 0 for i in range(k): ans += dp[i] * (k - i) ans %= MOD print(ans)