結果

問題 No.1897 Sum of 2nd Max
ユーザー LyricalMaestro
提出日時 2024-11-09 17:55:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 497 ms / 2,000 ms
コード長 883 bytes
コンパイル時間 360 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 75,904 KB
最終ジャッジ日時 2024-11-09 17:55:29
合計ジャッジ時間 8,584 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1897

MOD = 998244353

def my_pow(k, n):
    if k > 0:
        return pow(k, n, MOD)
    else:
        return 0

def main():
    N, K = map(int, input().split())

    answer = 0
    for k in reversed(range(1, K + 1)):
        # 一番大きい値がkの場合
        ans1 = my_pow(k, N)
        ans1 -= my_pow(k - 1, N)
        ans1 %= MOD
        ans1 -= (N * my_pow(k - 1, N - 1)) % MOD
        ans1 %= MOD

        # 一番大きい値がkより大きい場合
        ans2 = my_pow(k, N - 1)
        ans2 -= my_pow(k - 1, N - 1)
        ans2 %= MOD
        ans2 *= (K - k)
        ans2 %= MOD
        ans2 *= N
        ans2 %= MOD
#        print(f"k = {k}, ans1 = {ans1}, ans2 = {ans2}")

        ans = (ans1 + ans2) % MOD
        answer += (ans * k) % MOD
        answer %= MOD
    print(answer)








if __name__ == "__main__":
    main()
0