結果

問題 No.1330 Multiply or Divide
ユーザー c-yan
提出日時 2021-01-10 00:58:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 561 ms / 2,000 ms
コード長 794 bytes
コンパイル時間 220 ms
コンパイル使用メモリ 12,288 KB
実行使用メモリ 32,356 KB
最終ジャッジ日時 2025-06-20 01:26:12
合計ジャッジ時間 6,374 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff #

from heapq import heappop, heappush

INF = 10 ** 20

N, M, P, *A = map(int, open(0).read().split())

def f(x):
    c = 0
    while x % P == 0:
        c += 1
        x //= P
    return (x, c)

max_A = max(A)

if max_A > M:
    print(1)
    exit()

b = {}
for m, c in map(f, A):
    b.setdefault(c, 0)
    b[c] = max(b[c], m)
t = 1
for k in sorted(b):
    if b[k] <= t:
        del b[k]
    else:
        t = b[k]
cs = sorted(b)

if len(cs) == 0:
    print(-1)
    exit()

q = [(0, 1)]
d = {}
d[0] = 1
while q:
    c, x = heappop(q)
    if d[c] > x:
        continue
    if x * max_A > M:
        print(c + 1)
        break
    for k in cs:
        d.setdefault(c + 1 + k, 0)
        if x * b[k] > d[c + 1 + k]:
            d[c + 1 + k] = x * b[k]
            heappush(q, (c + 1 + k, x * b[k]))
0