結果

問題 No.473 和と積の和
コンテスト
ユーザー maspy
提出日時 2020-02-27 00:04:22
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 633 ms / 3,000 ms
コード長 695 bytes
コンパイル時間 92 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 47,276 KB
最終ジャッジ日時 2024-10-13 15:38:41
合計ジャッジ時間 28,253 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
# %%
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines


# %%
import numpy as np
from functools import lru_cache

# %%
N, X = map(int, read().split())
X += 1
# %%
x = np.arange(1, 10**5)
div = x[X % x == 0]
div = np.union1d(div, X // div)[1:].tolist()


# %%
@lru_cache(None)
def f(X, K, i):
    # i番目以上を使ってK個でXを作る
    if K == 1:
        return 1 if div[i] <= X else 0
    ret = 0
    for j, d in enumerate(div[i:], i):
        if d ** K > X:
            break
        if X % d != 0:
            continue
        ret += f(X // d, K - 1, j)
    return ret


# %%
print(f(X, N, 0))
0