[Programmers] 약수의 합 - 12928

728x90

sqrt()로 for문 연산 줄이는 코드 , List comprehension 코드 둘 다 기록하기!

sqrt() 연산시 제곱 수 중복 때문에 (ex. 16 = 4x4) list에 약수를 저장 후 set() 해줘야 한다!

 

Question

 

정수 n을 입력받아 n의 약수를 모두 더한 값을 리턴하는 함수, solution을 완성해주세요.

 

 

Point

 

1. Basic for문 : Code 1
2. List comprehension : Code 2
3. math.sqrt() 로 for문 연산 줄이기 : Code 3
4. set() Code 3번 제곱의 약수 중복 방지

 

 

Code 1 : Basic

 

def solution(n):
    sum=0
    for i in range(1,n+1):
        sum += i if n%i == 0 else 0
    return sum

 

 

Code 2 : List comprehension

 

def solution(n):
    return sum([i for i in range(1,n+1) if n%i ==0])

 

 

Code 3 : sqrt() 함수로 for문 연산 줄이기

 

import math
def solution(n):
    l1 = [i for i in range(1,int(math.sqrt(n))+1) if n%i == 0]
    l1.extend([n//i for i in l1])
    return sum(set(l1))

 

 

now me

On my Github