結果
問題 | No.1737 One to N |
ユーザー |
|
提出日時 | 2024-09-28 13:55:24 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 63 ms / 2,000 ms |
コード長 | 4,926 bytes |
コンパイル時間 | 213 ms |
コンパイル使用メモリ | 82,372 KB |
実行使用メモリ | 67,436 KB |
最終ジャッジ日時 | 2024-09-28 13:55:28 |
合計ジャッジ時間 | 3,105 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
import sysinput = sys.stdin.readlinesys.set_int_max_str_digits(0)from collections import defaultdict, deque, Counterfrom heapq import heappop, heappushfrom bisect import bisect_left, bisect_right## gcd(x, y):最大公約数, lcm(x, y):最小公倍数, factorial(n):階乗n!, prem(n, k):nPk(n, k), comb(n, r):二項係数nCrfrom math import gcd, lcm, factorial, perm, comb#0~9を並び替えるならpermutationsかconbinations,N列のカテゴリを作るにはproductfrom itertools import product, permutations, combinations, accumulatefrom functools import lru_cache #@lru_cache(maxsize=128)import operatorfrom string import ascii_uppercase, ascii_lowercase, digits # 英字(大文字), 英字(小文字), 数字MOD = 998244353def II():return int(input())def LI():return list(input())def LMI():return list(map(int, input().split()))def LMS():return list(map(str, input().split()))def LLMI(x):return [list(map(int, input().split())) for _ in range(x)]def LLMS(x):return [list(map(str, input().split())) for _ in range(x)]def CUM(x:list) -> list:'''func:累積の仕方を指定する。operator.mul:掛け算operator.sub:引き算max:最大値min:最小値initial:初期値, Noneならx[0]が第一引数の数値になる'''return list(accumulate(x, func=None, initial=0))def yn(tf:bool):if tf:return print('YES')else:return print('No')class UnionFind():def __init__(self, n):self.n = nself.parents = [-1] * ndef find(self, x):if self.parents[x] < 0:return xelse:self.parents[x] = self.find(self.parents[x])return self.parents[x]def union(self, x, y):x = self.find(x)y = self.find(y)if x == y:returnif self.parents[x] > self.parents[y]:x, y = y, xself.parents[x] += self.parents[y]self.parents[y] = xdef size(self, x):return -self.parents[self.find(x)]def same(self, x, y):return self.find(x) == self.find(y)def members(self, x):root = self.find(x)return [i for i in range(self.n) if self.find(i) == root]def roots(self):return [i for i, x in enumerate(self.parents) if x < 0]def group_count(self):return len(self.roots())def group(self):group_members = defaultdict(list)for member in range(self.n):group_members[self.find(member)].append(member)return group_membersdef __str__(self):return ''.join(f'{r}: {m}' for r, m in self.group().items())def inverse_element(num:int):'''逆元の作成ax ≡ 1 (mod p)となるxは、fetmatの小定理よりa * a^(p-2) ≡ 1 (mod p)であるため、a^(p-2) (mod p) は逆元である'''return pow(num, MOD-2, MOD)def make_graph(n:int, lmi:list, idx_0:bool):graph = [[] for _ in range(n)]for i in range(len(lmi)):a, b = lmi[i]if idx_0:a -= 1b -= 1# 有向グラフであれば1方向にappendする。graph[a].append(b)graph[b].append(a)return graphdef dfs(n:int, graph:list[list[int]], s:int = 0, g:int = None):'''s:start地点、指定しなければ頂点0からg:goal地点、指定しなければ端まで'''d = deque([(s, 0)])TF = [False] * nTF[s] = Truewhile d:crr, cnt = d.popleft()print(crr, cnt)if g is not None and crr == g:## gにたどり着けるかreturn Truefor nxt in graph[crr]:if TF[nxt]:continued.append((nxt, cnt+1))TF[nxt] = Trueelse:return Falsedef dijkstra(n:int, graph:list[list[int, int]], s:int = 0):'''s:start地点、指定しなければ頂点0から'''que = []heappush(que, (0, s))TF = [False] * n# 各頂点の最短経路を格納するans = [0] * nwhile que:cnt, crr = heappop(que)if TF[crr]: continue# 最短経路確定TF[crr] = Trueans[crr] = cntfor nxt, val in graph[crr]:# 最短経路が確定しているところは除くif TF[nxt]:continueheappush(que, (cnt+val, nxt))else:return ansdef prime_factorize(n):a = []while n % 2 == 0:a.append(2)n //= 2f = 3while f * f <= n:if n % f == 0:a.append(f)n //= felse:f += 2if n != 1:a.append(n)return adef execute():print(sum(prime_factorize(II())))if __name__ == "__main__":T = 1for _ in range(T):execute()