結果

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

ソースコード

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 = Counter({1: 1})
    while True:
        ok = False
        for i in range(len(primes)):
            if n % primes[i] == 0:
                if primes[i] in res:
                    res[primes[i]] += 1
                else:
                    res[primes[i]] = 1
                n = n // primes[i]
                break
            if primes[i] > n:
                ok = True
                break
        if ok:
            break
    return res


if __name__ == '__main__':
    N, H = map(int, input().split())
    A = list(map(int, 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