結果
問題 | No.2582 Random Average^K |
ユーザー | Navier_Boltzmann |
提出日時 | 2023-12-10 13:40:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,532 ms / 2,000 ms |
コード長 | 1,124 bytes |
コンパイル時間 | 332 ms |
コンパイル使用メモリ | 81,592 KB |
実行使用メモリ | 76,012 KB |
最終ジャッジ日時 | 2024-05-01 10:26:24 |
合計ジャッジ時間 | 9,454 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 32 ms
52,096 KB |
testcase_01 | AC | 31 ms
52,352 KB |
testcase_02 | AC | 33 ms
52,224 KB |
testcase_03 | AC | 32 ms
51,948 KB |
testcase_04 | AC | 31 ms
51,840 KB |
testcase_05 | AC | 32 ms
51,840 KB |
testcase_06 | AC | 52 ms
63,616 KB |
testcase_07 | AC | 44 ms
62,780 KB |
testcase_08 | AC | 59 ms
63,988 KB |
testcase_09 | AC | 471 ms
75,724 KB |
testcase_10 | AC | 684 ms
75,336 KB |
testcase_11 | AC | 711 ms
75,508 KB |
testcase_12 | AC | 1,532 ms
76,012 KB |
testcase_13 | AC | 1,388 ms
75,472 KB |
testcase_14 | AC | 34 ms
51,840 KB |
testcase_15 | AC | 1,521 ms
75,752 KB |
testcase_16 | AC | 846 ms
75,940 KB |
testcase_17 | AC | 882 ms
75,388 KB |
ソースコード
N,K = map(int,input().split()) mod = 998244353 # class combination(): # def __init__(self,N,p): # self.fact = [1, 1] # fact[n] = (n! mod p) # self.factinv = [1, 1] # factinv[n] = ((n!)^(-1) mod p) # self.inv = [0, 1] # factinv 計算用 # self.p = p # for i in range(2, N + 1): # self.fact.append((self.fact[-1] * i) % p) # self.inv.append((-self.inv[p % i] * (p // i)) % p) # self.factinv.append((self.factinv[-1] * self.inv[-1]) % p) # def cmb(self,n, r): # if (r < 0) or (n < r): # return 0 # r = min(r, n - r) # return self.fact[n] * self.factinv[r] * self.factinv[n-r] % self.p # C = combination(K+1+N,mod) ans = 0 C = 1 for i in range(N): if i%2==0: ans = (ans + C*pow(N-i,N+K,mod))%mod else: ans = (ans - C*pow(N-i,N+K,mod))%mod C = (N-i)*pow(i+1,mod-2,mod)*C%mod inv = 1 for i in range(K+1,N+K+1): inv = (inv*i)%mod inv = pow(inv,mod-2,mod) ans = (ans*inv)%mod inv = pow(pow(N,K,mod),mod-2,mod) print(ans*inv%mod)