[Lv. 1] 두 개 뽑아서 더하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/68644

프로그래머스

난이도 : Level.1

언어 : Kotlin

 

문제 설명 :

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

 

제한 사항 : 

• numbers의 길이는 2 이상 100 이하입니다.
  • numbers의 모든 수는 0 이상 100 이하입니다.

입출력 예 :

numbers	result
[2, 1, 3, 4, 1]	[2, 3, 4, 5, 6, 7]
[5, 0, 2, 7]	[2, 5, 7, 9, 12]

풀이 : 

class Solution {
    var answer: IntArray = intArrayOf()
    
    fun solution(numbers: IntArray): IntArray {
        combination(0, 2, 0, numbers, mutableListOf())
        return answer.distinct().sorted().toIntArray()
    }
    
    fun combination(cnt: Int, depth: Int, startAt: Int, arr: IntArray, temp: MutableList<Int>) {
        if (cnt == depth) {
            answer += temp.sum()
        } else {
            for (i in startAt until arr.size) {
                temp.add(arr[i])
                combination(cnt + 1, depth, i + 1, arr, temp)
                temp.removeAt(temp.lastIndex)
            }
        }
    }
}

 

뽑기 문제가 나오면 조합을 쓰는 편이라 조합을 쓴 뒤, 중복을 제거하고 정렬시켜 반환했다.

다만 이 방법 말고도 깔끔하게 스트림으로 풀어낸 다른 사람의 풀이가 존재하여 가져와봤다.

 

다른 사람의 풀이 : 

class Solution {
    fun solution(numbers: IntArray): IntArray {
        val list = numbers.toList()
        return list.withIndex().flatMap { i -> list.withIndex().map { j -> i to j } }
            .filter { it.first.index != it.second.index }
            .map { it.first.value + it.second.value }
            .toSortedSet()
            .toIntArray()
    }
}