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

https://programmers.co.kr/learn/courses/30/lessons/77884?language=kotlin 

코딩테스트 연습 - 약수의 개수와 덧셈

문제

문제 설명

두 정수 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 해야 합니다.

 

초기 코드

class Solution {
    fun solution(left: Int, right: Int): Int {
        var answer: Int = 0
        return answer
    }
}

의식의 흐름

1. 약수 세는법은 %해서 0이면 약수..

2. (left..right) 으로 포문 돌리면 되지 않을까?

 

해결 코드

class Solution {
    fun solution(left: Int, right: Int): Int {
        var answer = 0

        (left..right).forEach { num ->
            var count = 0

            (1..num).forEach {
                if (num % it == 0) count++
            }

            answer += when (count % 2) {
                0 -> num
                else -> -num
            }
        }

        return answer
    }
}