[Kotlin] 없는 숫자 더하기 (프로그래머스 알고리즘)

 

문제

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

 

 

---

 

접근

 

• 아래와 같은 방식으로 내부의 원소를 검사하고, 없는 경우 sum을 추가하면 된다고 생각했다.

class Solution {
    fun solution(numbers: IntArray): Int {
        
        var sum = 0
        if (numbers.contains(1) == false) sum += 1
        if (numbers.contains(2) == false) sum += 2
        if (numbers.contains(3) == false) sum += 3
        if (numbers.contains(4) == false) sum += 4
        if (numbers.contains(5) == false) sum += 5
        if (numbers.contains(6) == false) sum += 5
        .
        .
        .
        return sum
    }
}

• 이 내용을 for문으로 간단하게 정리하면 아래와 같이 완료된다.

 

---

 

성공 코드

 

class Solution {
    fun solution(numbers: IntArray): Int {
        var sum = 0
        for (i in 1..9) {
        if (numbers.contains(i) == false) sum += i }
        return sum
    }
}

 

---

 

설명

 

1. for문으로 1부터 9까지 반복하며 numbers 배열 안에 1부터 9까지의 원소가 있는지 검사한다.(contains)
2. 원소가 없는 경우(numbers.contains(i) == false) 에는 sum에 해당 원소를 더해준다.
3. 없는 원소가 모두 더해진 변수 sum을 리턴한다.
4. numbers.contains(i) == false 대신 i in numbers == false 로 사용할 수도 있다.

 

 

💡 Kotlin에서 배열 안의 특정 원소와 관련된 메서드

 

val array = arrayOf(1, 2, 3, 4, 5)
val result = 3 in array // true

• in 연산자
• 배열 안에 특정 원소가 있는지를 검사
• 해당 원소가 배열에 포함되어 있으면 true를 반환하고, 그렇지 않으면 false를 반환

 

val array = arrayOf(1, 2, 3, 4, 5)
val result = array.contains(3) // true

• contains() 
• 배열에 특정 원소가 포함되어 있는지를 검사
• 해당 원소가 배열에 포함되어 있으면 true를 반환하고, 그렇지 않으면 false를 반환

 

val array = arrayOf(1, 2, 3, 4, 5)
val containsElement = array.any { it == 3 } // true

• .any() 
• 배열에 주어진 조건을 만족하는 요소가 하나라도 있는지를 확인
• 만약 배열에 해당 조건을 만족하는 요소가 하나라도 있다면 true를 반환, 그렇지 않으면 false를 반환

 

val array = arrayOf(1, 2, 3, 4, 5)
val index = array.indexOf(3) // 2

• indexOf()
• 배열에서 특정 원소의 인덱스를 검색
• 해당 원소가 배열에 있으면 인덱스값, 배열에 없으면 -1을 반환

 

val sortedArray = arrayOf(1, 2, 3, 4, 5)
val index = sortedArray.binarySearch(3) // 2

• binarySearch()
• 배열이 정렬되어 있을 경우, 이진 검색을 수행하여 특정 원소의 인덱스를 검색
• 해당 원소가 배열에 있으면 인덱스값, 배열에 없으면 음수를 반환

 

val array = arrayOf(1, 2, 3, 4, 5)
val filteredArray = array.filter { it == 3 } // [3]
val containsElement = filteredArray.isNotEmpty() // true

• .filter() 
• 배열의 각 요소에 대해 주어진 조건을 만족하는 요소만을 선택하여 새로운 배열을 생성
• 따라서 특정 원소가 배열에 있는지를 확인하는 것보다는, 해당 조건을 만족하는 원소가 있는지를 확인하는 데 적합
  
  
  
  
  
  

 

---

 

다른 방법

 

class Solution {
    fun solution(numbers: IntArray): Int = (0..9).filterNot(numbers::contains).sum()
}

• 위와 반대로 .filter()가 아닌 .filterNot()을 사용한 방법이다.
• filterNot(numbers::contains)
  - 입력된 배열 numbers에 포함되지 않는 숫자들을 필터링
  - filterNot() : 주어진 조건을 만족하지 않는 요소들을 선택하여 새로운 컬렉션을 생성
  - numbers::contains : numbers의 contains() 메서드를 참조하는 메서드 참조
• sum()으로 이렇게 필터링된 숫자들의 합을 계산하고 리턴

 

class Solution {
    fun solution(numbers: IntArray): Int  = 45 - numbers.sum()
}

• 1부터 9까지 전부 더한 값(45)에서 numbers의 합을 빼준 간단한 풀이이다.