[Java] LeetCode 15번 3Sum

배열에서 3개의 수의 합이 0이되도록하는 수의 집합을 찾는 문제이다. 먼저 배열을 정렬한 뒤에 2개의 수를 고정하고, 나머지 하나를 binary search로 찾는 방식을 택했다. 시간복잡도는 O(n^2 log(2))이다. 

중복되는 집합을 피하기 위해서 전과 같은 수의 값이면 다음 값으로 넘어갈 수 있도록 continue 구문을 작성해주었다.

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> answer = new ArrayList<>();
        Arrays.sort(nums);
        int n = nums.length;
        for (int i = 0; i < n; i++) {
            if (i > 0 && nums[i] == nums[i - 1])
                continue;
            for (int j = i + 1; j < n; j++) {
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                int left = j + 1;
                int right = n - 1;
                while (left <= right) {
                    int mid = left + (right - left) / 2;
                    int sum = nums[i] + nums[j] + nums[mid];
                    if (sum > 0) {
                        right = mid - 1;
                    } else if (sum < 0) {
                        left = mid + 1;
                    } else {
                        List<Integer> sumSet = new ArrayList<>();
                        sumSet.add(nums[i]);
                        sumSet.add(nums[j]);
                        sumSet.add(nums[mid]);
                        answer.add(sumSet);
                        break;
                    }
                }
            }
        }
        return answer;
    }
}