3Sum LeetCode 解题报告

题目

Given an array S of n integers, are there elements a, b, c in S 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.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

找出数组中三个数相加为0的数，不能是重复的数字组合。找三个数的和为0，可以依次取数组中一个数设为target，然后在后面找两个数，使得这两个数的和为-target，这样找出来的数相加和就为0了。

需要注意的地方：
（1）、首先对数组排序，方便后面找数字和。
（2）、使用首尾标记，分别标记target后的首尾数字nums[low]和nums[high]，当nums[i+1] == nums[i]时，需要跳过,执行low++，因为这时会有重复的数字组合。同理nums[high-1] == nums[high]时，执行high–，从而避免找到重复的数字组合。
（3）、当找到的两个数和sum与-target相加不为0时，比较sum和-taget的大小，sum > -target时，说明后面两个数相加大了，high++（把和变小，这也是为什么先排序），如果sum < -target，说明后面两个数的和小了，那么low++，把后面两个数和增大。

代码和简要解析

public class Solution {    public List<List<Integer>> threeSum(int[] nums) {        List<List<Integer>> list = new ArrayList<List<Integer>>();        Arrays.sort(nums);        if(nums.length < 3) return list;        for(int i = 0; i < nums.length-2; i++){            if(i == 0 || (i > 0 && nums[i] != nums[i-1])){  //这是为了避免重复的三元组，i>0避免数组越界                int low = i+1;                      int high = nums.length-1;                int sum = -nums[i];         //取一个数的负数设为target，后面去两个数相加与target的和看是否为0                while(low < high){                    if(nums[low]+nums[high] == sum){                        list.add(Arrays.asList(nums[i],nums[low],nums[high]));                        while(low < high && nums[low] == nums[low+1]) low++;        //避免重复组合                        while(low < high && nums[high] == nums[high-1]) high--;     //避免重复组合                        low++;                        high--;                    }else if(nums[low]+nums[high] < sum){                        low++;                    }else{                        high--;                    }                }            }        }        return list;    }}