結果

問題 No.1897 Sum of 2nd Max
ユーザー siman
提出日時 2022-11-27 22:01:36
言語 Ruby
(3.4.1)
結果
AC  
実行時間 790 ms / 2,000 ms
コード長 277 bytes
コンパイル時間 46 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 12,416 KB
最終ジャッジ日時 2024-10-04 13:18:56
合計ジャッジ時間 13,803 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

N, K = gets.split.map(&:to_i)
MOD = 998_244_353

def f(x)
  q(x) - q(x + 1)
end

def q(x)
  return 0 if x > K

  (K.pow(N, MOD) - (x - 1).pow(N, MOD) - (K - x + 1) * N * (x - 1).pow(N - 1, MOD)) % MOD
end

ans = 0

1.upto(K) do |i|
  ans += i * f(i)
  ans %= MOD
end

puts ans
0