15. 三数之和

链接：https://leetcode-cn.com/problems/3sum/

 

中文题目：

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。注意：答案中不可以包含重复的三元组。

 

英文题目：

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.

 

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，满足要求的三元组集合为：

[

[-1, 0, 1],

[-1, -1, 2]

]

 

思路：

将数组排好序，先确定下一个数字，从这个数字右边第一位为第二个数，排序数组最后一个数为第三个输。这样做的好处是，第二个数只能往右边找，第三个数只能往左边找。

 

 

题解：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        // 先将数组排好序，可将时间复杂度减少至O(n*n);
        Arrays.sort(nums);

        // 只需判断n - 2位即可
        for (int i = 0; i < nums.length - 2; ++i) {
            // 从第二位开始，若与第一位相同，根据不重复原则去重
            if (i > 0 && nums[i - 1] == nums[i]) {
                continue;
            }
            // 第二位数从i + 1开始，第三位数从最末尾开始
            // 当第二位不满足时，只需要向右调大即可，同理第三位向左调小
            int second = i + 1, third = nums.length - 1;
            while (second < third) {
                int answer = nums[i] + nums[second] + nums[third];
                // 满足条件
                if (answer == 0) {
                    List<Integer> resTemp = new ArrayList<>();
                    resTemp.add(nums[i]);
                    resTemp.add(nums[second]);
                    resTemp.add(nums[third]);
                    result.add(resTemp);
                    // 加入时，将第二位第三位同时位移，避免出现相同的数字
                    ++second;
                    --third;

                    // 去重需去干净，防治出现连续相等的数出现
                    while (second < third && nums[second - 1] == nums[second]) {
                        ++second;
                    }
                    while (second < third && nums[third] == nums[third + 1]) {
                        --third;
                    }
                } else if (answer < 0) {
                    // 三数之和小于0，则将第二个数调大再判断，毕竟第三个数没办法调大了
                    ++second;
                } else {
                    // 同理
                    --third;
                }
            }
        }
        return result;
    }
}