프로그래머스 삼총사 문제 바로가기

간만에 푼 코딩테스트 문제. 문제 조건에 따라서 3개씩 뽑아 삼총사의 조건(세 수의 합이 0)을 만족하는지 확인해주면 간단히 해결 가능하다. Python의 경우 collections의 combination을 이용해 sum이 0인지 확인하면 더 편할듯하다. (메모리 초과는 차치해두면)

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> number) {
    int answer = 0;
    for (int i=0; i<number.size()-2; i++){
        for (int j=i+1; j<number.size()-1; j++){
            for (int k=j+1; k<number.size(); k++){
                if (number[i] + number[j] + number[k] == 0){
                    answer++;
                }
            }
        }
    }
    return answer;
}