leetcode 18. 四数之和【双指针】

三数之和的基础上理解这题的设计思路不难,个人认为难点在于去重的处理。
下面以数组[0,0,0,0,0,1,2,3,3,3]举例说明:

  1. 第一层遍历的去重:当数字第一次等于0时,此时不应该判断去重,因为题目没有限制后面的三个数字不能重复,如果第一次等于0时去重,则后面的数字不能取到0。所以 nums[i] == nums[i-1] 的判断条件保证了在一连串的重复数字中,第一次取到该数字时不会去重。
  2. 第二层遍历同理,后面两个数字可以与第二个数字重复,第一次取某值时不应去重。
  3. 最后两个数字的去重,比较简单,用while循环即可实现。
class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        
        for(int i=0; i<nums.length-3;i++){
            if(i>0 && nums[i] == nums[i-1]){continue;}
            int target1 = target - nums[i];
            for(int j=i+1; j<nums.length-2;j++){
                if(j>i+1 && nums[j] == nums[j-1]){continue;}
                int target2 = target1 - nums[j];
                int k=j+1;
                int l = nums.length-1;
                while(k<l){
                    if(nums[k] + nums[l] == target2){ 
                        res.add(Arrays.asList(nums[i], nums[j], nums[k], nums[l]));
                        while(k<l && nums[k] == nums[k+1])k++;
                        while(k<l && nums[l] == nums[l-1])l--;
                        k++;
                        l--;
                    }else if(nums[k] + nums[l] < target2){
                        k++;                        
                    }else{
                       l--; 
                    }                    
                }
            }
        }
        return res;
    }
}
发布了55 篇原创文章 · 获赞 0 · 访问量 785

猜你喜欢

转载自blog.csdn.net/er_ving/article/details/104777513
今日推荐