結果
| 問題 | No.3651 K-th Sum of Divisors |
| コンテスト | |
| ユーザー |
まぬお
|
| 提出日時 | 2026-08-28 22:02:19 |
| 言語 | PyPy3 (7.3.17) |
| 結果 |
AC
|
| 実行時間 | 280 ms / 2,000 ms |
| + 183µs | |
| コード長 | 1,209 bytes |
| 記録 | |
| コンパイル時間 | 247 ms |
| コンパイル使用メモリ | 96,240 KB |
| 実行使用メモリ | 165,076 KB |
| 最終ジャッジ日時 | 2026-08-28 22:02:37 |
| 合計ジャッジ時間 | 15,345 ms |
|
ジャッジサーバーID (参考情報) |
judge2_0 / judge1_0 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 55 |
ソースコード
from collections import deque, defaultdict, Counter
from bisect import bisect_left, bisect_right, insort
from itertools import permutations, combinations, groupby
from heapq import heappop, heappush
import math, sys
input = lambda: sys.stdin.readline().rstrip("\r\n")
def printl(li, sep=" "): print(sep.join(map(str, li)))
def yn(flag): print(Yes if flag else No)
_int = lambda x: int(x)-1
MOD = 100003
INF = 1<<60
Yes, No = "Yes", "No"
# B_i <- A_x_i
def _f(x, p): # O(NlogP) 使いまわすなら真面目に
n = len(x)
if p == 0:
return [i for i in range(n)]
elif p%2:
y = _f(x, p-1)
return [y[v] for v in x]
else:
y = _f(x, p//2)
return [y[v] for v in y]
def div(num):
left = []
right = []
i = 1
while i*i <= num:
if num%i == 0:
left.append(i)
if i*i < num: right.append(num//i)
i += 1
ret = left + right[::-1]
return ret
def f(n):
d = div(n)
v = sum(d)%MOD
return v
N, K = map(int, input().split())
if K == 1:
print(N)
exit()
P = [0]*(MOD+1)
v = f(N)
u = N
if N >= MOD: u = MOD
P[u] = v
for n in range(MOD):
v = f(n)
P[n] = v
Q = _f(P, K-1)
print(Q[u])
まぬお