結果

問題 No.2365 Present of good number
ユーザー detteiuu
提出日時 2025-03-04 04:41:04
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 73 ms / 2,000 ms
コード長 1,495 bytes
コンパイル時間 718 ms
コンパイル使用メモリ 82,652 KB
実行使用メモリ 74,680 KB
最終ジャッジ日時 2025-03-04 04:41:09
合計ジャッジ時間 5,158 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

from heapq import heappush, heappop

def factorization(n):
    arr = []
    temp = n
    for i in range(2, int(-(-n**0.5//1))+1):
        if temp%i == 0:
            cnt = 0
            while temp%i == 0:
                cnt += 1
                temp //= i
            arr.append([i, cnt])
    if temp != 1:
        arr.append([temp, 1])
    if arr == []:
        arr.append([n, 1])
    return arr

def matrix(a, b):
    ans = [[0]*len(b[0]) for _ in range(len(a))]
    for i in range(len(a)):
        for j in range(len(b[0])):
            ans[i][j] = sum(a[i][k]*b[k][j]%(MOD-1) for k in range(len(b)))%(MOD-1)
    return ans

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

MOD = 10**9+7

que = []
S = set()
for n, c in factorization(N):
    heappush(que, -n)
    S.add(n)

while que:
    a = -heappop(que)
    for n, c in factorization(a+1):
        if n not in S:
            heappush(que, -n)
            S.add(n)
S = sorted(S)
D = dict()
for i, s in enumerate(S):
    D[s] = i

dp = [[[0]*len(D) for _ in range(len(D))]]
for i, s in enumerate(S):
    n = s+1
    for j, s2 in enumerate(S):
        cnt = 0
        while n%s2 == 0:
            n //= s2
            cnt += 1
        dp[0][j][i] = cnt

for _ in range(59):
    dp.append(matrix(dp[-1], dp[-1]))

A = [[0] for _ in range(len(D))]
for n, c in factorization(N):
    A[D[n]][0] = c

for i in range(60):
    if 1<<i & K:
        A = matrix(dp[i], A)

ans = 1
for i in range(len(D)):
    ans *= pow(S[i], A[i][0], MOD)
    ans %= MOD

print(ans)
0