結果
| 問題 |
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)