結果
問題 | No.1554 array_and_me |
ユーザー |
![]() |
提出日時 | 2025-01-22 03:07:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 767 ms / 2,000 ms |
コード長 | 655 bytes |
コンパイル時間 | 441 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 111,600 KB |
最終ジャッジ日時 | 2025-01-22 03:07:34 |
合計ジャッジ時間 | 22,618 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
import sys input = sys.stdin.readline from heapq import heappop,heappush mod=998244353 F=[1] for i in range(1,200020): F.append(F[-1]*i%mod) T=int(input()) for tests in range(T): N,K=map(int,input().split()) A=list(map(int,input().split())) SUM=sum(A) INV=pow(SUM,mod-2,mod) H=[] for a in A: H.append((-a,-a*INV%mod,1)) H.sort() ANS=1 for i in range(K): #print(H) now,x,ko=heappop(H) ANS=ANS*(-x)%mod y=x*ko*pow(ko+1,mod-2,mod)%mod heappush(H,(now*ko/(ko+1),y,ko+1)) #print(ANS) for i in range(1,K+1): ANS=ANS*i%mod print(ANS)