結果

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

ソースコード

diff #

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)

    
0