leetcode打卡day4

三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1],[-1, -1, 2]]

原来解法(通过了311/313个测试用例,超出执行时间):

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        Set<List<Integer>> result = new LinkedHashSet<>();
        int i,j,k;
        for(i=0;i<nums.length-2;i++){
            if(nums[i]>0)  break;
            for(j=i+1;j<nums.length-1;j++){
                for(k=j+1;k<nums.length;k++){
                    if(nums[i]+nums[j]+nums[k]==0){
                        List<Integer> value = Arrays.asList(nums[i], nums[j], nums[k]);
                    result.add(value);
                    }
                }
            }
        }
        return new ArrayList<>(result);
    }
}

改进后的解法(参考题解):

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
       Arrays.sort(nums);
       List<List<Integer>> result=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;
            int j=i+1,k=nums.length-1;
            while(j<k){
                int sum=nums[i]+nums[j]+nums[k];
                if(sum==0){
                result.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],nums[k])));
                    while(j<k&&nums[j]==nums[++j]);
                    while(j<k&&nums[k]==nums[--k]);
                    }
                    else if(sum>0){
                        while(j<k&&nums[k]==nums[--k]);
                    }
                    else{
                         while(j<k&&nums[j]==nums[++j]);
                    }
                }
            }
        return result;
    }
}

思路:先通过sort对数组从小到大排序,nums[i]如果是正数就终止,然后用双指针j和k,从两端向中间运行,如果三数之和小于0,左指针j向右移动一位;三数之和大于0,右指针k向左移动一位;等于0的话则同时移动一位。

上一篇:C++ Day4 指针和结构体


下一篇:Python学习Day4