leetcode 15.三数之和 java

题目：三数之和

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

注意：答案中不可以包含重复的三元组。

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

---

分析：

说实话，我第一反应就是按照我人工做道题的思路，三层循环一个个找，去重用的是Set，而且我想不到别的什么好办法了。

代码：

public List<List<Integer>> threeSum(int[] nums) {
        if (nums.length<3) return new ArrayList<>();
        //将数组排序，并封装为List，为了使用contaion()函数
        Arrays.sort(nums);
        List<Integer> numList= Arrays.stream(nums).boxed().collect(Collectors.toList());
        //寻找符合条件的组合加入set
        Set<List<Integer>> set=new HashSet<>();
        for(int i=0;i<numList.size() && nums[i]<=0;i++){
            for (int j=i+1;j<numList.size() ;j++){
                List<Integer> list=new ArrayList<>();
                if (numList.subList(j+1,numList.size()).contains(-nums[i]-nums[j])){
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(-nums[i]-nums[j]);
                    set.add(list);
                }
            }
        }

        return new ArrayList<>(set);
}

虽然自己测试能通过，但在leetcode上判定的是运行超时，大约还是运行时间太长了。
 

学习：

看的这位大佬的题解（link）。

准备工作是排除掉不可能出结果的数组和重复的数字。
真正的算法是使用双指针代替了两层循环，通过和的大小判断指针的移动方向（并且每次移动后都跳过重复数字，不做重复计算）。

很厉害的一点是对重复数字的跳过，使用这样的代码：

while (left<right && nums[left]==nums[++left]);//注意：是++left，不是left++

相当于可以连续判断多个数字是不是重复数字，就很灵性，很厉害。
 
代码：

public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res=new ArrayList<>();
        for (int i=0;i<nums.length-2;i++){
            if (nums[i]>0) break;
            if (i>0 && nums[i]==nums[i-1]) continue;//注意：i>0，要防止取nums[i-1]不超出数组界限
            int left=i+1,right=nums.length-1;
            while (left<right){
                int sum=nums[i]+nums[left]+nums[right];
                if (sum<0){
                    while (left<right && nums[left]==nums[++left]);
                }else if (sum>0){
                    while (left<right && nums[right]==nums[--right]);
                }else {
                    res.add(new ArrayList<>(Arrays.asList(nums[i],nums[left],nums[right])));
                    while(left < right && nums[left] == nums[++left]);
                    while(left < right && nums[right] == nums[--right]);
                }
            }
        }
        return res;
}

总结：

1. 连续跳过多个重复数字：

while nums[i]==nums[++i]);//注意：是++i，不是i++

2. 初始化List集合（以ArrayList为例）：
   ArrayList有三个构造函数，一是空构造函数，二是以初始容量作为参数，三是以一个集合作为参数：public ArrayList(Collection<? extends E> c) 。
   实现Collection接口的有很多集合类，常用的主要有List、Set、Queue、Stack，因此在给List初始化的时候，参数注意是一个集合类实例，可以将数组转为集合，但参数不能直接是数组或数字。
   如：

new ArrayList<>(Arrays.asList(1,2,3));

3. Arrays.sort()是将原数组重新排序，从小到大，时间复杂度是O(nlogn)。
4. 取数组元素注意不要超出数组前（当取当前数字的前面数字时）后（当取当前数字的后面数字时，数组长度）界限。