結果

問題 No.1897 Sum of 2nd Max
ユーザー LyricalMaestroLyricalMaestro
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
52,352 KB
testcase_01 AC 32 ms
51,968 KB
testcase_02 AC 33 ms
51,968 KB
testcase_03 AC 402 ms
75,624 KB
testcase_04 AC 451 ms
75,904 KB
testcase_05 AC 219 ms
75,544 KB
testcase_06 AC 231 ms
75,648 KB
testcase_07 AC 284 ms
75,648 KB
testcase_08 AC 256 ms
75,776 KB
testcase_09 AC 407 ms
75,776 KB
testcase_10 AC 393 ms
75,648 KB
testcase_11 AC 419 ms
75,520 KB
testcase_12 AC 401 ms
75,392 KB
testcase_13 AC 398 ms
75,520 KB
testcase_14 AC 135 ms
75,428 KB
testcase_15 AC 170 ms
75,392 KB
testcase_16 AC 196 ms
75,776 KB
testcase_17 AC 160 ms
75,500 KB
testcase_18 AC 181 ms
75,392 KB
testcase_19 AC 34 ms
52,096 KB
testcase_20 AC 32 ms
52,096 KB
testcase_21 AC 34 ms
52,352 KB
testcase_22 AC 34 ms
52,480 KB
testcase_23 AC 34 ms
52,096 KB
testcase_24 AC 33 ms
52,224 KB
testcase_25 AC 33 ms
51,968 KB
testcase_26 AC 33 ms
51,840 KB
testcase_27 AC 33 ms
52,096 KB
testcase_28 AC 33 ms
51,840 KB
testcase_29 AC 35 ms
52,096 KB
testcase_30 AC 425 ms
75,544 KB
testcase_31 AC 418 ms
75,612 KB
testcase_32 AC 338 ms
75,520 KB
testcase_33 AC 497 ms
75,648 KB
権限があれば一括ダウンロードができます

ソースコード

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