結果

問題 No.368 LCM of K-products
ユーザー 🍡yurahuna
提出日時 2016-04-11 22:17:14
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 569 ms / 2,000 ms
コード長 1,087 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-09-22 14:41:35
合計ジャッジ時間 10,070 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict as ddict

# sqrt(10^9) < 40000未満の素数をあらかじめ列挙しておく (4203個)
prime = []
memo = [True] * 40000
i = 2
while i < 40000:
    if memo[i]:
        prime.append(i)
        j = 2 * i
        while j < 40000:
            memo[j] = False
            j += i
    i += 1

# print(len(prime))

###########################

N, K = map(int, input().split())
a = list(map(int, input().split()))

exponents = ddict(list)
for ax in a:
    # 素因数分解(作成済みの素数表primeを使う)
    cnt = ddict(int)
    for p in prime:
        while ax % p == 0:
            ax //= p
            cnt[p] += 1

    # sqrt(10^9) 以下の素数で割り切れなかったら、a_iは素数
    if ax != 1:
        cnt[ax] += 1

    for key, value in cnt.items():
        exponents[key].append(value)

# print(exponents)

mod = 10**9 + 7

ans = 1
for prime, li in exponents.items():
    li.sort(reverse = True)
    num = 0
    for i in range(min(K, len(li))):
        num += li[i]

    ans *= pow(prime, num, mod)
    ans %= mod

print(ans)
0