結果
問題 | No.1897 Sum of 2nd Max |
ユーザー |
|
提出日時 | 2021-11-20 01:42:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 368 ms / 2,000 ms |
コード長 | 344 bytes |
コンパイル時間 | 882 ms |
コンパイル使用メモリ | 82,196 KB |
実行使用メモリ | 80,988 KB |
最終ジャッジ日時 | 2024-11-28 11:32:51 |
合計ジャッジ時間 | 6,704 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
def solve(n, k): q = [0] * (k + 10) mod = 998244353 ans = 0 for i in range(k + 1, 0, -1): q[i] = pow(k, n, mod) - pow(i - 1, n, mod) - (k - i + 1) * n * pow(i - 1, n - 1, mod) q[i] %= mod ans += i * (q[i] - q[i + 1]) ans %= mod return ans n, k = map(int,input().split()) print(solve(n, k))