当前位置：首页 > article >正文

Leetcode 18-四数之和

article 2025/2/26 3:54:39

题解看代码随想录

/*
1.先排序，为了便于使用双指针，复杂度O(logN)
2.固定最小指针a,b,双指针c和d分别从0和len-1向中间移动
从两端向中间移动是为了使得两个数下标不重合，且可以将时间复杂度从双重循环的O(N^4)降到O(N^3)
3.终止条件：a>N-3||nums[a]>0(此时四数之和必定大于0)
3.不可以包含重复元素：前后元素相同时跳过
*/
class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        int sum=0,c,d;
        List<List<Integer>> res = new ArrayList<>();
        for(int a=0;a<nums.length-3;a++){
            //提前终止
            //不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了
            if(nums[a]>target&&nums[a]>=0) break;
            //对a进行去重
            if(a>0&&nums[a]==nums[a-1]) continue;
            for(int b=a+1;b<nums.length-2;b++){
                //多判断b>a+1是为了避免在[2,2,2,2,2],target=8时没有输出
                //所以不判断b=a+1的情况，即b可以等于a
                if(b>a+1&&nums[b]==nums[b-1]) continue;
                c=b+1;
                d=nums.length-1;
                while(c<d){
                    sum=nums[a]+nums[b]+nums[c]+nums[d];
                    if(sum>target){
                        d--;
                    }else if(sum<target){
                        c++;
                    }else{
                        res.add(Arrays.asList(nums[a],nums[b],nums[c],nums[d]));
                        while(c<d&&nums[c]==nums[c+1]) c++;
                        while(c<d&&nums[d]==nums[d-1]) d--;
                        c++;
                        d--;

                    }

                }
            }
        }

        return res;
    }
}