문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/131705
문제 설명
한국중학교에 다니는 학생들은 각자 정수 번호를 갖고 있습니다. 이 학교 학생 3명의 정수 번호를 더했을 때 0이 되면 3명의 학생은 삼총사라고 합니다. 예를 들어, 5명의 학생이 있고, 각각의 정수 번호가 순서대로 -2, 3, 0, 2, -5일 때, 첫 번째, 세 번째, 네 번째 학생의 정수 번호를 더하면 0이므로 세 학생은 삼총사입니다. 또한, 두 번째, 네 번째, 다섯 번째 학생의 정수 번호를 더해도 0이므로 세 학생도 삼총사입니다. 따라서 이 경우 한국중학교에서는 두 가지 방법으로 삼총사를 만들 수 있습니다.
한국중학교 학생들의 번호를 나타내는 정수 배열 number
가 매개변수로 주어질 때, 학생들 중 삼총사를 만들 수 있는 방법의 수를 return 하도록 solution 함수를 완성하세요.
제한 사항
- 3 ≤
number
의 길이 ≤ 13
- -1,000 ≤
number
의 각 원소 ≤ 1,000
- 서로 다른 학생의 정수 번호가 같을 수 있습니다.
문제 풀이
번호 쌍은 순서와 무관하게 요소가 같다면 동일한 쌍이다. 즉 [1, 2, 3]
과 [3, 2, 1]
은 동일한 쌍이다.
순서에 따라 평가값이 달라지는 것이 아니기 때문에, 단순 순회로 각 쌍을 평가하고 0이라면 return할 값을 증가하는 것으로 구현하면 된다.
동일한 쌍을 중복 순회하지 않도록 for문의 초기식과 조건식에 주목하자.
소스 코드
public int solution(int[] number)
{
int length = number.Length;
if (length < 3 || length > 13)
{
return 0;
}
int answer = 0;
for (int i = 0; i < length - 2; i++)
{
for (int j = i + 1; j < length - 1; j++)
{
int sum = number[i] + number[j];
for (int k = j + 1; k < length; k++)
{
if (sum + number[k] == 0)
{
answer += 1;
}
}
}
}
return answer;
}