데이터분석 기록일지

문제풀이/프로그래머스

[프로그래머스 Lv.1] 약수의 합 (Python)

야하루 2024. 7. 8. 19:56

코딩테스트 연습 - 약수의 합 | 프로그래머스 스쿨 (programmers.co.kr)

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

문제

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

 

 

제한 사항

n은 0 이상 3000이하인 정수입니다.

 

입출력 예
n return
12 28
5 6

 

 


 

풀이

 

아이디어: 약수이므로 범위를 1~n으로 두고, 그 수로 n을 나눴을 때 나머지가 0인 값들을 찾아 더한다.

 

# 풀이1
def solution(n):
    answer = 0
    for i in range(1,n+1):
        if n%i == 0:
            answer += i 
    return answer

 

 

# 풀이2
def solution(n):
    answer = sum([x for x in range(1,n+1) if n%x==0])
    return answer

 

 

-> 둘 다 방법은 똑같고 for문과 if 문을 다 쓰느냐, 리스트 컴프리헨션을 쓰느냐의 차이이다.