[Algorithm] 삼총사

문제 설명

한국중학교에 다니는 학생들은 각자 정수 번호를 갖고 있습니다.

이 학교 학생 3명의 정수 번호를 더했을 때 0이 되면 3명의 학생은 삼총사라고 합니다.

예를 들어, 5명의 학생이 있고, 각각의 정수 번호가 순서대로 -2, 3, 0, 2, -5일 때, 첫 번째, 세 번째, 네 번째 학생의 정수 번호를 더하면 0이므로 세 학생은 삼총사입니다. 또한, 두 번째, 네 번째, 다섯 번째 학생의 정수 번호를 더해도 0이므로 세 학생도 삼총사입니다. 따라서 이 경우 한국중학교에서는 두 가지 방법으로 삼총사를 만들 수 있습니다. 한국중학교 학생들의 번호를 나타내는 정수 배열 number가 매개변수로 주어질 때, 학생들 중 삼총사를 만들 수 있는 방법의 수를 return 하도록 solution 함수를 완성하세요.

 

문제 유형

• 조합

 

의사코드

1. [반복문] 벡터 길이만큼 반복한다. (초기값, a  = 0)
   1. [반복문] 벡터 길이만큼 반복한다. (초기값, b = a + 1)
      1. [반복문] 벡터 길이만큼 반복한다. (초기값, c = b + 1) 
         1. 인덱스 a,b,c에 해당하는 벡터 요소를 모두 더해 0인지 확인한다.
            1. 0일 경우 result + 1한다.
2. 결과값(result)를 반환한다.

 

코드 작성

#include <vector>

using namespace std;

int solution(vector<int> number) {
    int result = 0;
    int size = number.size();
    
    for (int a = 0; a < size; a++)
    {
        for (int b = a + 1; b < size; b++)
        {
            for (int c = b + 1; c < size; c++)
            {
                if ((number[a] + number[b] + number[c]) == 0) result++;
            }
        }
    }
    
    return result;
}