15. 3Sum

题目要求结果里不能出现重复的结果。

Example 1:

Input: nums = [-1,0,1,2,-1,-4]
Output: [[-1,-1,2],[-1,0,1]]
Explanation: 
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0.
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0.
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0.
The distinct triplets are [-1,0,1] and [-1,-1,2].
Notice that the order of the output and the order of the triplets does not matter.

Example 2:

Input: nums = [0,1,1]
Output: []
Explanation: The only possible triplet does not sum up to 0.

Example 3:

Input: nums = [0,0,0]
Output: [[0,0,0]]
Explanation: The only possible triplet sums up to 0.

Constraints:

  • 3 <= nums.length <= 3000

  • -105 <= nums[i] <= 105

My Solutions:

先把nums排序。这样在遍历数组时可以跳过重复数字。对每一个遍历的数字,在ta的后面的数字中用双指针进行搜索。

Time: O(N^2)

Space: O(N)

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length - 2; i++) {
            if (i == 0 || nums[i] != nums[i - 1]) { // 跳过重复数字
                int l = i + 1, r = nums.length - 1, sum = -nums[i];
                while (l < r) {
                    if (nums[l] + nums[r] == sum) {
                        res.add(Arrays.asList(nums[i], nums[l], nums[r]));
                        while (l < r && nums[l] == nums[l + 1]) l++; // skip the duplicate
                        while (l < r && nums[r] == nums[r - 1]) r--; // skip the duplicate
                        l++; // skip one more time to make sure the next number is instinct
                        r--; // skip one more time to make sure the next number is instinct
                    }
                    else if (nums[l] + nums[r] < sum) l++;
                    else r--;
                }
            }
        }
        return res;
    }
}

Last updated