結果

問題 No.1554 array_and_me
ユーザー tktk_snsntktk_snsn
提出日時 2021-06-18 23:33:27
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 955 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 81,972 KB
実行使用メモリ 124,244 KB
最終ジャッジ日時 2024-06-22 21:33:20
合計ジャッジ時間 32,915 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 125 ms
89,936 KB
testcase_01 AC 620 ms
96,124 KB
testcase_02 AC 570 ms
96,100 KB
testcase_03 AC 565 ms
95,572 KB
testcase_04 AC 576 ms
96,288 KB
testcase_05 AC 575 ms
96,428 KB
testcase_06 WA -
testcase_07 AC 1,192 ms
123,220 KB
testcase_08 AC 1,224 ms
123,876 KB
testcase_09 AC 1,217 ms
124,244 KB
testcase_10 AC 1,181 ms
122,172 KB
testcase_11 AC 341 ms
113,996 KB
testcase_12 AC 326 ms
113,956 KB
testcase_13 AC 332 ms
113,980 KB
testcase_14 AC 338 ms
113,888 KB
testcase_15 AC 330 ms
113,848 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 AC 869 ms
120,900 KB
testcase_37 AC 853 ms
121,024 KB
testcase_38 AC 848 ms
120,536 KB
testcase_39 AC 869 ms
119,136 KB
testcase_40 AC 846 ms
121,264 KB
testcase_41 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from fractions import Fraction
import heapq
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
mod = 998244353
U = 10 ** 5 + 10

fact = [1] * (U+1)
inv = [1] * (U+1)
for i in range(1, U+1):
    fact[i] = fact[i-1] * i % mod
inv[U] = pow(fact[i], mod-2, mod)
for i in reversed(range(U)):
    inv[i] = inv[i+1] * (i + 1) % mod


T = int(input())
for _ in range(T):
    N, K = map(int, input().split())
    A = list(map(int, input().split()))
    S = sum(A) % mod
    S_inv = pow(S, mod-2, mod)

    que = [(-Fraction(a), i) for i, a in enumerate(A)]
    heapq.heapify(que)
    cnt = [0] * N
    for _ in range(K):
        f, idx = heapq.heappop(que)
        f = -f
        num, div = f.as_integer_ratio()
        cnt[idx] += 1
        heapq.heappush(que, (-Fraction(num, div + 1), idx))

    res = fact[K]
    for i in range(N):
        res = res * pow(A[i] * S_inv, cnt[i], mod)
        res = res * inv[cnt[i]] % mod
    print(res)
0