해당 풀이는 파이썬 알고리즘 인터뷰 책을 참고하였습니다.
Given an integer array nums, return all the triplets [nums[i], nums[j], nums[k]] such that i != j, i != k, and j != k, and nums[i] + nums[j] + nums[k] == 0. Notice that the solution set must not contain duplicate triplets.
public List<List<Integer>> threeSum2(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
Arrays.sort(nums);
int left, right, sum;
for (int i = 0; i < nums.length - 2; i++) {
// 중복값 건너뛰기
if (i > 0 && nums[i] == nums[i - 1]) continue;
// i를 기준으로 left는 i 바로 왼쪽부터, right는 배열의 끝부터
left = i + 1;
right = nums.length - 1;
// 배열의 세 자리의 합을 구해서 0이 나오는 지 확인. left는 right를 넘어갈 수 없다.
while (left < right) {
sum = nums[i] + nums[left] + nums[right];
if (sum < 0) {
left += 1;
} else if (sum > 0) {
right -= 1;
} else {
list.add(Arrays.asList(nums[i], nums[left], nums[right]));
// 중복된 값을 피하기 위해 left가 right를 넘어가지 않는 조건 하에 앞 자리와 같은 수가 있으면 스킵
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
// 마지막으로 한 칸씩 더 옮겨 온다(마찬가지로 중복값을 피하기 위해).
left++;
right--;
}
}
}
return list;
}
ㅋㅋ3sum