프로그래머스 - 삼총사

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

프로그래머스

 

조합 알고리즘을 사용하는 전형적인 문제다. 조합을 이용해서 3개를 뽑았을때 3개의 합이 0 인 경우면 answer 값을 증가시켜주면 된다.

class Solution {
    var max = 3 // 뽑을 숫자 개수
    lateinit var result: Array<Int> // 뽑은 결과
    var answer: Int = 0
    fun solution(number: IntArray): Int {
        result = Array(max) {0}
        combination(number,0,0)
        return answer
    }
    
    fun combination(number: IntArray ,index: Int, count: Int){
        if(count == max){
            if(result.sum() == 0){
                answer++
            }
            return
        }

        for(i in index until number.size){
            result[count] = number[i]
            combination(number,i + 1, count + 1)
        }
    }
}