結果

問題 No.1554 array_and_me
ユーザー hirayuu_yc
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0