結果

問題 No.2798 Multiple Chain
ユーザー shotoyooshotoyoo
提出日時 2024-06-28 22:21:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 855 ms / 2,000 ms
コード長 3,314 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,184 KB
実行使用メモリ 215,624 KB
最終ジャッジ日時 2024-06-28 22:22:39
合計ジャッジ時間 46,634 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 791 ms
215,240 KB
testcase_01 AC 808 ms
215,424 KB
testcase_02 AC 813 ms
215,452 KB
testcase_03 AC 793 ms
215,524 KB
testcase_04 AC 819 ms
215,308 KB
testcase_05 AC 801 ms
215,424 KB
testcase_06 AC 845 ms
215,288 KB
testcase_07 AC 794 ms
215,424 KB
testcase_08 AC 793 ms
215,344 KB
testcase_09 AC 809 ms
215,304 KB
testcase_10 AC 795 ms
215,424 KB
testcase_11 AC 823 ms
215,452 KB
testcase_12 AC 792 ms
215,424 KB
testcase_13 AC 804 ms
215,232 KB
testcase_14 AC 788 ms
215,092 KB
testcase_15 AC 815 ms
215,296 KB
testcase_16 AC 802 ms
215,424 KB
testcase_17 AC 807 ms
215,304 KB
testcase_18 AC 796 ms
215,552 KB
testcase_19 AC 782 ms
215,496 KB
testcase_20 AC 796 ms
215,308 KB
testcase_21 AC 804 ms
215,424 KB
testcase_22 AC 802 ms
215,344 KB
testcase_23 AC 803 ms
215,296 KB
testcase_24 AC 798 ms
215,424 KB
testcase_25 AC 806 ms
215,296 KB
testcase_26 AC 809 ms
215,552 KB
testcase_27 AC 807 ms
215,348 KB
testcase_28 AC 797 ms
215,588 KB
testcase_29 AC 800 ms
215,116 KB
testcase_30 AC 809 ms
215,296 KB
testcase_31 AC 817 ms
215,248 KB
testcase_32 AC 800 ms
215,456 KB
testcase_33 AC 795 ms
215,528 KB
testcase_34 AC 817 ms
215,296 KB
testcase_35 AC 789 ms
215,452 KB
testcase_36 AC 792 ms
215,252 KB
testcase_37 AC 793 ms
215,236 KB
testcase_38 AC 793 ms
215,264 KB
testcase_39 AC 791 ms
215,264 KB
testcase_40 AC 855 ms
215,364 KB
testcase_41 AC 840 ms
215,552 KB
testcase_42 AC 824 ms
215,372 KB
testcase_43 AC 851 ms
215,360 KB
testcase_44 AC 818 ms
215,424 KB
testcase_45 AC 799 ms
215,180 KB
testcase_46 AC 799 ms
215,320 KB
testcase_47 AC 795 ms
215,424 KB
testcase_48 AC 787 ms
215,624 KB
testcase_49 AC 791 ms
215,176 KB
testcase_50 AC 798 ms
215,296 KB
testcase_51 AC 789 ms
215,320 KB
testcase_52 AC 796 ms
215,424 KB
testcase_53 AC 817 ms
215,456 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys, random
input = lambda : sys.stdin.readline().rstrip()



write = lambda x: sys.stdout.write(x+"\n"); writef = lambda x: print("{:.12f}".format(x))
debug = lambda x: sys.stderr.write(x+"\n")
YES="Yes"; NO="No"; pans = lambda v: print(YES if v else NO); INF=10**18
LI = lambda v=0: list(map(lambda i: int(i)-v, input().split())); II=lambda : int(input()); SI=lambda : [ord(c)-ord("a") for c in input()]
def debug(_l_):
    for s in _l_.split():
        print(f"{s}={eval(s)}", end=" ")
    print()
def dlist(*l, fill=0):
    if len(l)==1:
        return [fill]*l[0]
    ll = l[1:]
    return [dlist(*ll, fill=fill) for _ in range(l[0])]

from math import gcd
import random
def is_prime(n):
    """miller_rabinによる素数判定
    ※ 1は素数と扱う
    """
    l = [2,3,5,7,11,13,17,19,23,29,31,37]
    if n==1 or n in l:
        return True
    d = n-1
    s = 0
    while d%2==0:
        s += 1
        d //= 2
    for a in l:
        v = pow(a,d,n)
        if v==1 or v==n-1:
            continue
        for _ in range(s):
            v = v*v % n
            if v==n-1:
                break
        else:
            return False
    return True
def rho(n):
    """nを割り切る3以上の素数を返す(素数のときnを返す)
    """
    if is_prime(n):
        return n
    while True:
        x = y = random.randint(1,n-1)
        g = 1
        while g==1:
            x = (x*x - 3) % n
            y = (y*y - 3) % n
            y = (y*y - 3) % n
            g = gcd((x-y), n)
        if g>1:
            return rho(g)
def factor(n):
    """高速な素因数分解
    """
    if n==1:
        return {}
    f = is_prime(n)
    if f:
        return {n:1}
    ans = {}
    while n%2==0:
        ans.setdefault(2, 0)
        ans[2] += 1
        n //= 2
    v = rho(n)
    while v!=n and n>1:
        ans.setdefault(v, 0)
        while n%v==0:
            n //= v
            ans[v] += 1
        if n>3 and is_prime(n):
            ans.setdefault(n,0)
            ans[n] += 1
            return ans
        v = rho(n)
    if n>1:
        ans.setdefault(n, 0)
        ans[n] += 1
    return ans

n = int(input())
m = 62
dp = dlist(m,m,m,m) # dp[v0][v][l][k] := 末尾v, 長さl, 総和k の単調非減少列の個数
for v in range(m):
    dp[v][v][1][v] = 1
for v in range(m):
    for l in range(1,m):
        for k in range(m):
            for v0 in range(m):
                val = dp[v0][v][l][k]
                if val==0:
                    continue
                for nv in range(v,m):
                    nk = k + nv
                    nl = l+1
                    if nk>=m or nl>=m:
                        break
                    dp[v0][nv][nl][nk] += val

f = factor(n)
ks = list(f.keys())
vs = list(f.values())
from itertools import product
ans = 0
for t in product(*[range(val) for val in vs]):
    # ks[i] が t[i] の状態
    cur = 1
    L = INF
    for i in range(len(t)):
        cur *= ks[i]**t[i]
        if t[i]>0:
            L = min(L, vs[i]//t[i])
    if L==INF:
        continue
    for l in range(1,L+1):
        res = 1
        for i in range(len(t)):
            tmp = 0
            for vv in range(m):
                tmp += dp[t[i]][vv][l][vs[i]]
            res *= tmp
            if res==0:
                break
        ans += res
print(ans+1)
0