結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0