結果
問題 | No.1554 array_and_me |
ユーザー |
![]() |
提出日時 | 2021-06-18 23:15:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 499 ms / 2,000 ms |
コード長 | 1,315 bytes |
コンパイル時間 | 155 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 92,544 KB |
最終ジャッジ日時 | 2024-06-22 21:24:52 |
合計ジャッジ時間 | 15,417 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
mod = 998244353eps = 10**-9def main():import sysfrom heapq import heappop, heappushinput = sys.stdin.readline# comb initnmax = 10 ** 5 + 10 # change herefac = [0] * nmaxfinv = [0] * nmaxinv = [0] * nmaxfac[0] = 1fac[1] = 1finv[0] = 1finv[1] = 1inv[1] = 1for i in range(2, nmax):fac[i] = fac[i - 1] * i % modinv[i] = mod - inv[mod % i] * (mod // i) % modfinv[i] = finv[i - 1] * inv[i] % moddef comb(n, r):if n < r:return 0else:return (fac[n] * ((finv[r] * finv[n - r]) % mod)) % modfor _ in range(int(input())):N, K = map(int, input().split())A = list(map(int, input().split()))S = sum(A)B = [0] * Npq = []for i in range(N):heappush(pq, (-A[i], i))for _ in range(K):v, i = heappop(pq)B[i] += 1heappush(pq, (-A[i] * (1 / (B[i] + 1)), i))S_inv = pow(S, mod-2, mod)T = Kans = 1for i in range(N):ans = (ans * pow((A[i] * S_inv)%mod, B[i], mod))%modans = (ans * comb(T, B[i]))%modT -= B[i]print(ans)assert T == 0if __name__ == '__main__':main()