사용자 도구

사이트 도구


ps:problems:programmers:12928

약수의 합

ps
링크programmers.co.kr/…
출처프로그래머스
문제 번호12928
문제명약수의 합
레벨Level 1
분류

수학

시간복잡도O(sqrt(n))
인풋사이즈n <= 3000
사용한 언어Python
해결날짜2021/06/08

풀이

  • 어떤 수의 모든 약수의 합을 구하는 것은, 그냥 모든 약수를 다 구해서 더하는 것이 가장 평범하고 빠르다.
  • i를 1부터 sqrt(n)까지 증가시키면서, n이 i로 나누어 떨어지면 i와 n/i 는 모두 n의 약수이다.
    • i가 sqrt(n)일 경우에는 i와 n/i 가 같으므로 중복해서 더하지 않도록 주의할것.
  • 시간 복잡도는 O(sqrt(n)).

코드

"""Solution code for "Programmers 12928. 약수의 합".

- Problem link: https://programmers.co.kr/learn/courses/30/lessons/12928
- Solution link: http://www.teferi.net/ps/problems/programmers/12928
"""

import math


def solution(n):
    sqrt = math.isqrt(n)
    answer = sum((i + n // i) for i in range(1, sqrt + 1) if n % i == 0)
    if sqrt * sqrt == n:
        answer -= sqrt
    return answer

토론

댓글을 입력하세요:
Z I S C᠎ G
 
ps/problems/programmers/12928.txt · 마지막으로 수정됨: 2021/06/10 14:18 저자 teferi