코딩 테스트/Python

[프로그래머스 코딩테스트 연습] 9. 약수의 합

알밤바 2022. 2. 22. 15:13
728x90
반응형
 

코딩테스트 연습 - 약수의 합

정수 n을 입력받아 n의 약수를 모두 더한 값을 리턴하는 함수, solution을 완성해주세요. 제한 사항 n은 0 이상 3000이하인 정수입니다. 입출력 예 n return 12 28 5 6 입출력 예 설명 입출력 예 #1 12의 약수

programmers.co.kr


1) 문제 설명

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

 

2) 제한 사항

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

 

3) 입출력 예 설명

입출력 예 #1
12의 약수는 1, 2, 3, 4, 6, 12입니다. 이를 모두 더하면 28입니다.

입출력 예 #2
5의 약수는 1, 5입니다. 이를 모두 더하면 6입니다.

 


1. 문제 접근 방식

i가 n의 약수라면 n을 i로 나누었을 때 나머지가 없어야 한다. 이것을 바탕으로 코드를 짜보았다.

 

2. 풀이코드

(1) 나의 풀이코드

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

 

(2) 다른 사람 풀이코드

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

리스트 컴프리헨션으로 푼 코드이다. 위의 코드와는 동일하게 진행되었지만 그것을 리스트 컴프리헨션으로 짠 코드다.

간단히 한 줄로 작성할 수 있는 점이 좋다. 

 

3. 배운 점

range에서 다소 막혔다. 하지만 이번 문제를 통해 명확히 깨달았으며, 인덱스에서 마지막 인덱스는 항상 원하는 값 + 1이란 것을 까먹지 말자.

어느정도 if, for문을 사용하는 방법을 알게 된 것 같다. 그래서 다른 사람이 풀었던 코드처럼 '리스트 컴프리헨션'을 활용해 코드를 짜는 연습을 해봐야겠다.

728x90
반응형