結果
| 問題 | No.3651 K-th Sum of Divisors |
| コンテスト | |
| ユーザー |
detteiuu
|
| 提出日時 | 2026-08-28 21:45:50 |
| 言語 | PyPy3 (7.3.17) |
| 結果 |
AC
|
| 実行時間 | 240 ms / 2,000 ms |
| + 892µs | |
| コード長 | 1,757 bytes |
| 記録 | |
| コンパイル時間 | 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 |
ソースコード
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)
detteiuu