結果
問題 | No.1554 array_and_me |
ユーザー |
![]() |
提出日時 | 2021-06-18 23:13:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 340 ms / 2,000 ms |
コード長 | 1,427 bytes |
コンパイル時間 | 170 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 100,552 KB |
最終ジャッジ日時 | 2024-06-22 21:23:36 |
合計ジャッジ時間 | 11,242 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
SIZE=10**5+10; MOD=998244353 #ここを変更するinv = [0]*SIZE # inv[j] = j^{-1} mod MODfac = [0]*SIZE # fac[j] = j! mod MODfinv = [0]*SIZE # finv[j] = (j!)^{-1} mod MODfac[0] = fac[1] = 1finv[0] = finv[1] = 1for i in range(2,SIZE):fac[i] = fac[i-1]*i%MODfinv[-1] = pow(fac[-1],MOD-2,MOD)for i in range(SIZE-1,0,-1):finv[i-1] = finv[i]*i%MODinv[i] = finv[i]*fac[i-1]%MODdef choose(n,r): # nCk mod MOD の計算if 0 <= r <= n:return (fac[n]*finv[r]%MOD)*finv[n-r]%MODelse:return 0def chofuku(box,ball): # nHk mod MOD の計算if box == ball == 0: return 1return choose(box+ball-1,ball)def perm(n,r): # nPr mod MODif 0 <= r <= n:return fac[n]*finv[n-r]%MODelse:return 0from heapq import *def solve(n,k,a):sa = sum(a)a.sort(reverse=1)pa = [ai/sa for ai in a]# 確率、回数、indexq = [(-ai,0,i) for i,ai in enumerate(pa)]for _ in range(k):p,c,idx = heappop(q)p = -pa[idx]/(c+2)heappush(q,(p,c+1,idx))ans = pow(sa,MOD-1-k,MOD)*fac[k]%MODfor _,c,i in q:ans *= pow(a[i],c,MOD)ans %= MODans *= finv[c]ans %= MODreturn ansimport sysreadline = sys.stdin.readlineT = int(input())for _ in range(T):n,k = map(int,readline().split())*a, = map(int,readline().split())ans = solve(n,k,a)print(ans)