結果
問題 |
No.1554 array_and_me
|
ユーザー |
![]() |
提出日時 | 2021-06-18 23:32:55 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 983 bytes |
コンパイル時間 | 378 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 48,324 KB |
最終ジャッジ日時 | 2024-06-22 21:32:43 |
合計ジャッジ時間 | 10,517 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 5 TLE * 1 -- * 35 |
ソースコード
from fractions import Fraction import heapq from itertools import count 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)