LeetCode P18---四数之和 三数之和的plus

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

此题和三数之和最接近的三数之和类似

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
注意:
答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

方法是二重for循环加上双指针法

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
       Arrays.sort(nums);
        List<List<Integer>> rs = new ArrayList<>();
        int len = nums.length;
        for (int i=0;i<len;i++){
            for (int j=i+1;j<len;j++){
                int left = j+1,right = len-1;
                while (left<right){
                    int sum = nums[i]+nums[j]+nums[left]+nums[right];
                    if (sum<target){
                        while (left<right&&nums[left]==nums[++left]);
                    }else if (sum>target){
                        while (left<right&&nums[right]==nums[--right]);
                    }else {
                        rs.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],nums[left],nums[right])));
                        while (left<right&&nums[left]==nums[++left]);
                        while (left<right&&nums[right]==nums[--right]);
                    }
                }
            }
        }
        //利用HashSet去除重复元素
        HashSet hashSet = new HashSet(rs);
        rs.clear();
        rs.addAll(hashSet);
        return rs;
    }
}

上面的用到了HashSet,虽然能达到目的且能成功AC,但是可以进一步优化,在for循环里面就可以进行去除重复元素,减少运行时间和空间
如下:

public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> rs = new ArrayList<>();
        if (nums==null||nums.length<4){
            return rs;
        }
        int len = nums.length;
        //第一层for循环不遍历全部,因为至少要留三个数给后面的,不然不符合题意
        for (int i=0;i<len-3;i++){
        	//跳过重复元素,避免出现重复结果
            if (i>0&&nums[i]==nums[i-1]) continue;
            //同上,至少要留两数给后面的
            for (int j=i+1;j<len-2;j++){
            	//跳过重复元素,避免出现重复结果
                if (j>i+1&&nums[j]==nums[j-1]) continue;
                int left = j+1,right = len-1;
                //双指针
                while (left<right){
                    int sum = nums[i]+nums[j]+nums[left]+nums[right];
                    if (sum<target){
                    //sum比目标小,左指针右移
                    //同时跳过重复元素,避免出现重复结果
                        while (left<right&&nums[left]==nums[++left]);
                    }else if (sum>target){
                    //sum比目标大,右指针左移
                    //同时跳过重复元素,避免出现重复结果
                        while (left<right&&nums[right]==nums[--right]);
                    }else {
                    //sum和target相等就加入到结果列表
                        rs.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],nums[left],nums[right])));
                        //跳过重复元素
                        while (left<right&&nums[left]==nums[++left]);
                        while (left<right&&nums[right]==nums[--right]);
                    }
                }
            }
        }
        return rs;
    }

两次执行的结果对比
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值