LeetCode | 3Sum

15. 3Sum

Description

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? 
Find all unique triplets in the array which gives the sum of zero.

Note:
The solution set must not contain duplicate triplets.

Example:
Given array nums = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

Solution: (Java)

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);

        int left, right;
        for (int i = 0; i < nums.length-2; i++) {
            left = i + 1;
            right = nums.length - 1;

            if (i == 0 || nums[i] != nums[i-1]) {
                while (left < right) {
                    int sum = nums[i] + nums[left] + nums[right];

                    if (sum == 0) {
                        List<Integer> tmp = new ArrayList<>();
                        tmp.add(nums[i]);
                        tmp.add(nums[left]);
                        tmp.add(nums[right]);
                        result.add(tmp);

                        left++;
                        right--;

                        while (left < right && nums[right] == nums[right+1])
                            right--;
                        while (left < right && nums[left] == nums[left-1])
                            left++;

                    } else if (sum > 0) {
                        right--;        // 要先减1，然后再循环判断，少这一步可能会死循环；上面和下面同理
                        while (left < right && nums[right] == nums[right+1])
                            right--;
                    } else {
                        left++;
                        while (left < right && nums[left] == nums[left-1])
                            left++;
                    }
                }
            }

        }
        return result;
    }
}

思路

这道题如果直接用 3 层 for 循环来判断三个数的和肯定是超时的，这里参考了其他网友的一些思路，先对数组排序，通过先固定一个数，然后找另外两个数。找另外两个数的时候（即相当于两数之和），通过首尾两个指针（left、right）向中间移动的方法，这样是一个线性的时间复杂度，提高效率。还有就是，因为数组排过序，在3个指针移动的时候可以跳过重复的数字，进行剪枝。此外，还有一个可以提高一些效率的地方，上文代码没加进去了，就是第一个固定的数 nums[i] > 0时，可以直接跳出循环（因为数组排过序）。