LeetCode第十五题-找出数组中三数和为0的答案

3Sum
问题简介:
给定n个整数的数组nums,是否有元素a,b,c在nums中,使a + b + c = 0?
找到数组中所有唯一的三元组,它们的总和为零
注:解决方案集不得包含重复的三元组
例如{1,-1,0}和{1,0,-1}

举例:
给定数组 {-2,1,0,-1,-1,2,3}
输出{
{1,0,-1},
{-1,-1,2},
{-2,0,2},
{-2,-1,3}
}
注:不能{0,0,0}

错误解法:Time Limit Exceeded
三层遍历导致时间复杂度极高

class Solution {
    public List> threeSum(int[] nums) {
        Set> set = new HashSet<>();
        int length = nums.length;
        if(length < 3) return new ArrayList();
        for(int i = 0;i list = Arrays.asList(num);
                        set.add(list);
                    }
                }
            }
        }
        List result = new ArrayList<>(set);
        return result;
    }
}

正确解法:
可以先将数组排序,做一个判断即排序数组第一个数字大于0或最后一个数字小于0返回空结果
遍历一遍数组,将0与第一个数字做差,差值在剩下的数组里寻找是否有两个数的合值符合要求

class Solution {
    public List> threeSum(int[] nums) {
        Arrays.sort(nums);
        List> result = new ArrayList();
        int length = nums.length;
        if(length ==0 || nums[0]>0 || nums[length-1]<0){
        return result;
        }
         for (int k = 0; k < length; ++k) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            int target = 0 - nums[k];
            int i = k + 1, j = length - 1;
            while (i < j) {
                if (nums[i] + nums[j] == target) {
                    List list = new ArrayList();
                    list.add(nums[k]);
                    list.add(nums[i]);
                    list.add(nums[j]);
                    result.add(list);
                    while (i < j && nums[i] == nums[i + 1]) ++i;
                    while (i < j && nums[j] == nums[j - 1]) --j;
                    ++i; --j;
                } else if (nums[i] + nums[j] < target) ++i;
                else --j;
            }
        }
        return result;
}
}

注:list和set之间转换可以通过初始化
例如已有一个set集合想将其转为list
List list = new ArrayList(set);

小白刷题之路,请多指教— — 要么大器晚成,要么石沉大海

你可能感兴趣的:(leetcode)