[Kotlin] 약수의 개수와 덧셈 (프로그래머스 알고리즘)

 

 

 

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/77884

 

 

 

---

 

접근

 

• 아래와 같이 구상했는데 ArithmeticException 예외가 났다.

접근, "main" java.lang.ArithmeticException 발생


class Solution {
    fun solution(left: Int, right: Int): Int {
        
        var answer = 0
        for(i in left .. right){  // left부터 right까지 반복
            
            var count = 0
            for(v in 0..i){
                if(i % v == 0) count++
            }  // 약수의 개수를 구하는 식
            
            if(count % 2 == 0) answer += i else answer -= i
            // 약수의 개수가 짝수면 answer에 더하고, 짝수면 뺌
                                    
        }
        return answer
        // answer를 리턴
    }
}

• for(v in 0..i){ if( i % v == 0) count ++} 에서 i를 0으로 나누는 경우가 있어서 생긴 문제다.
• 1로 수정하니 해결되었다.

 

---

 

성공 코드

 

class Solution {
    fun solution(left: Int, right: Int): Int {
        
        var answer = 0
        for(i in left .. right){  // left부터 right까지 반복
            var count = 0
            for(v in 1..i){
                if(i % v == 0) count++
            }  // 약수의 개수를 구하는 식
            
            if(count % 2 == 0) answer += i else answer -= i
            // 약수의 개수가 짝수면 answer에 더하고, 짝수면 뺌
                                    
        }
        return answer
        // answer를 리턴
    }
}

 

---

 

설명

 

1. left 수부터 right 수까지 반복하는 for문을 만든다.
2. for문 안에서 약수의 개수를 구하고
3. 개수가 짝수인지 홀수인지 판단해 answer에 더하거나 빼주는 작업을 한다.
4. answer를 리턴한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(left: Int, right: Int): Int {
        return (left..right).map { i -> if ((1..i).filter { i % it == 0 }.size % 2 == 0) i else -i }.sum()
    }
}

• map과 filter, 람다함수를 이용한 풀이다.
• (left..right)
  left부터 right까지의 범위를 생성한다. 이 범위에 포함된 모든 정수들을 반복하면서 처리할 것이다.
• .map { i -> ... }
  생성된 범위의 각 요소에 대해 람다 함수를 적용한다. 각 요소를 i라는 이름의 변수로 받아들이고, 람다 함수 내에서 처리할 것이다.
• (1..i).filter { i % it == 0 }
  1부터 i까지의 범위에서 i를 나누었을 때 나머지가 0인 요소들을 필터링한다. 즉, i의 약수들을 구한다.
• .size
  약수의 개수를 구한다.
• % 2 == 0
  약수의 개수가 짝수인지 확인한다. 짝수이면 참(true)을 반환하고, 홀수이면 거짓(false)을 반환한다.
• if ... i else ... -i
  약수의 개수가 짝수이면 i를 반환하고, 홀수이면 -i를 반환한다.
• .sum()
  선택된 값들의 합을 계산한다.