結果

問題 No.1250 汝は倍数なりや?
コンテスト
ユーザー ryhoh
提出日時 2020-11-16 22:31:13
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 1,787 bytes
コンパイル時間 492 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 413,932 KB
最終ジャッジ日時 2024-07-23 01:46:18
合計ジャッジ時間 5,789 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29 RE * 1 TLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import Counter
from functools import reduce
from operator import add
import math


def make_primes(size: int) -> list:
    if not isinstance(size, int):
        raise AttributeError("args must be int but given %s" % size.__class__)
    if size < 0:
        raise ValueError("args must be (size < 0) but given %s" % size)
    if size < 2:
        return []

    res = []
    flag = [True for _ in range(size + 1)]
    limit = int(math.sqrt(size))

    for i in range(2, limit + 1):
        if flag[i]:
            res.append(i)
            for j in range(i, size + 1, i):
                flag[j] = False

    for i in range(max(2, limit), size + 1):
        if flag[i]:
            res.append(i)

    return res


def prime_factor(n: int, primes: list) -> Counter:
    if not isinstance(n, int) or not isinstance(primes, list):
        raise AttributeError("args must be int but given (%s, %s)" % (n.__class__, primes.__class__))
    if n < 1:
        raise ValueError("args must be (n < 1) but given %s" % n)

    res = {1: 1}
    ok = False
    while not ok:
        for prime in primes:
            if n % prime == 0:
                if prime in res:
                    res[prime] += 1
                else:
                    res[prime] = 1
                n = n // prime
                break
            if prime > n:
                ok = True
                break
        else:
            break

    return Counter(res)


if __name__ == '__main__':
    N, H = map(int, input().split())
    A = [abs(int(i)) for i in input().split()]

    primes = make_primes(H+1)
    A_factor = reduce(add, (prime_factor(a, primes) for a in A))
    h_factor = prime_factor(H, primes)

    if A_factor & h_factor == h_factor:
        print("YES")
    else:
        print("NO")
0