結果

問題 No.3651 K-th Sum of Divisors
コンテスト
ユーザー detteiuu
提出日時 2026-08-28 21:45:50
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 240 ms / 2,000 ms
+ 892µs
コード長 1,757 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,261 ms
コンパイル使用メモリ 95,716 KB
実行使用メモリ 132,480 KB
最終ジャッジ日時 2026-08-28 21:46:11
合計ジャッジ時間 17,011 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge2_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

class Eratosthenes:
    def __init__(self, n):
        self.isPrime = [True]*(n+1)
        self.minfactor = [-1]*(n+1)
        self.isPrime[0], self.isPrime[1] = False, False
        self.minfactor[1] = 1
        for i in range(2, n+1):
            if self.isPrime[i]:
                self.minfactor[i] = i
                for j in range(i*i, n+1, i):
                    self.isPrime[j] = False
                    if self.minfactor[j] == -1:
                        self.minfactor[j] = i
    
    def factorize(self, n):
        factor = []
        while n > 1:
            p = self.minfactor[n]
            cnt = 0
            while self.minfactor[n] == p:
                n //= p
                cnt += 1
            factor.append((p, cnt))
        return factor
    
    def divisor(self, n):
        ans = [1]
        pf = self.factorize(n)
        for p, c in pf:
            L = len(ans)
            for i in range(L):
                v = 1
                for _ in range(c):
                    v *= p
                    ans.append(ans[i]*v)
        return ans

MOD = 100003
E = Eratosthenes(10**6*3)

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

if K == 1:
    exit(print(N))

IDX = [-1]*MOD
G = [-1]*MOD
A = [-1, N, sum(E.divisor(N))%MOD]
if N < MOD:
    IDX[N] = 1
if A[-1] == A[-2]:
    exit(print(A[-1]))
IDX[A[-1]] = 2
if N < MOD:
    G[N] = A[-1]
if K == 2:
    exit(print(A[-1]))

start = -1
while True:
    nex = sum(E.divisor(A[-1]))%MOD
    nidx = IDX[A[-1]]+1
    if IDX[nex] == -1:
        IDX[nex] = nidx
        G[A[-1]] = nex
        if IDX[nex] == K:
            exit(print(nex))
        A.append(nex)
    else:
        start = nex
        K = (K-IDX[nex])%(nidx-IDX[nex])
        break

for _ in range(K):
    start = G[start]

print(start)
0