結果
問題 |
No.1554 array_and_me
|
ユーザー |
|
提出日時 | 2025-04-17 21:50:30 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 481 bytes |
コンパイル時間 | 272 ms |
コンパイル使用メモリ | 82,456 KB |
実行使用メモリ | 112,376 KB |
最終ジャッジ日時 | 2025-04-17 21:50:58 |
合計ジャッジ時間 | 23,988 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 19 WA * 22 |
ソースコード
from fractions import Fraction from heapq import* MOD=998244353 T=int(input()) for _ in range(T): N,K=map(int,input().split()) A=list(map(int,input().split())) hq=[-Fraction(A[i],1) for i in range(N)] heapify(hq) ans=pow(sum(A),-K,MOD) for i in range(K): nf=heappop(hq) ans*=(i+1) ans*=-nf.numerator ans*=pow(nf.denominator,-1,MOD) ans%=MOD heappush(hq,Fraction(nf.numerator,nf.denominator+1)) print(ans)