알고리즘

[프로그래머스] 약수의 개수와 덧셈_Python

작은코딩 2022. 5. 2. 09:49

🔒 문제

문제 설명

두 정수 left와 right가 매개변수로 주어집니다. left부터 right까지의 모든 수들 중에서, 약수의 개수가 짝수인 수는 더하고, 약수의 개수가 홀수인 수는 뺀 수를 return 하도록 solution 함수를 완성해주세요.


제한사항
  • 1 ≤ left  right ≤ 1,000

입출력 예

left right result
13 17 43
24 27 52

입출력 예 설명

입출력 예 #1

  • 다음 표는 13부터 17까지의 수들의 약수를 모두 나타낸 것입니다.
약수 약수의 개수
13 1, 13 2
14 1, 2, 7, 14 4
15 1, 3, 5, 15 4
16 1, 2, 4, 8, 16 5
17 1, 17 2
  • 따라서, 13 + 14 + 15 - 16 + 17 = 43을 return 해야 합니다.

입출력 예 #2

  • 다음 표는 24부터 27까지의 수들의 약수를 모두 나타낸 것입니다.
약수 약수의 개수
24 1, 2, 3, 4, 6, 8, 12, 24 8
25 1, 5, 25 3
26 1, 2, 13, 26 4
27 1, 3, 9, 27 4
  • 따라서, 24 - 25 + 26 + 27 = 52를 return 해야 합니다.

🔓 풀이

🔑 요구사항 파악

1. 요구사항

 - 매개변수로 주어지는 두 정수 사이의 모든 수의 약수 개수를 구하고 약수 개수가 짝수면 덧셈, 홀수면 뺄셈을 해서 리턴

 

2. Type

 - Input 

left: int, right: int

 - Output 

answer: int

🔑 문제 정의

1. 약수 구하기


🔑 문제 해결 방법 고민하기

1. 약수 구하기
 - 1. 1부터 n까지 모든 수로 나눌 때 나누어 떨어지는 수가 약수
 - 2. 어짜피 제곱근이 정수가 아니라면 약수는 항상 짝수(제곱수라면 홀수, 아니면 짝수)

1-2.
 - 제곱근을 구한뒤 정수인지 아닌지 판단해서 정수라면 -, 정수가 아니라면 +

 

<고민의 흔적>


🔑 문제 해결 / 코드

def solution(left, right):
    answer = 0
    for i in range(left, right + 1):
        square_root = i**(1/2)
        if int(str(square_root).split(".")[1]) == 0:
            answer -= i
        else:
            answer += i
    return answer


left = 13
right = 17

solution(left, right)

💉 피드백

1. 제곱근 구하기

제곱근은 i**(1/2) or i**0.5 식으로 구할 수 있다.

 

2. 제곱근이 정수인지 아닌지 판별하기 

나는 문자열로 바꿔서 "."을 기준으로 슬라이싱 후 소수점 아래 자리가 0인지 아닌지 구분하는 것으로 조건문을 만들었는데 실수를 정수로 만들어서 일치하는지 여부를 판단하는게 좀 더 좋은 코드로 보인다.

def solution(left, right):
    answer = 0
    for i in range(left, right + 1):
        square_root = i**(1/2)
        ####
        if int(square_root) == square_root:
            answer -= i
        else:
            answer += i
        ####
    return answer