삼총사

문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/92341?language=kotlin

프로그래머스

문제

나의 풀이

정수가 몇가지 주어졌을때 그중에서 3개를 고르고 그 수들의 합이 0이 되면 삼총사가 되는 간단한 조합 문제이다.

순열 (Permutation) & 조합 (Combination)

이전에 정리 해놓았던 게시글을 참고해서 쉽게 구현했다.

class Solution {
    var num = ArrayList<Int>()
    val pickednum = mutableListOf<Int>()
    var count = 0
    fun solution(number: IntArray): Int {
        for(i in number.indices) num.add(number[i])
        return combination(0, 3, 0)
    }
    
    fun combination(cnt: Int, depth: Int, beginWith: Int): Int{
        if(cnt == depth) {
            if(pickednum.sum() == 0) {
                println(pickednum)
                count++ 
            }
        }
        for(i in beginWith until num.size) {
            pickednum.add(num[i])
            combination(cnt + 1, depth, i + 1)
            pickednum.removeAt(pickednum.lastIndex)
        }
        return count
    }
}