LeetCode 四数之和 Java 双指针

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

注意:答案中不可以包含重复的四元组

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum

相似题目

三数之和

最接近的三数之和

思路

这题只是三数之和稍加改一下
三数之和是由i的位置来确定a,然后left从i+1开
始来确定b,right从nums.length-1开始来确定c;

四数之和则多加一重循环,同样i的位置来确定a,但
i+1的位置来确定b,left从j+1的位置来确定c,right从nums.length-1开始来确定d
 
剩下的就都差不多,包括去重的方法,跟三数之和的方法是一样的直接看代码

代码

class Solution {
    
    
    public List<List<Integer>> fourSum(int[] nums, int target) {
    
    
    List<List<Integer>> ans=new ArrayList();
    if(nums.length==0||nums.length<4)
    return  ans;
    if(nums.length==4&&(nums[0]+nums[1]+nums[2]+nums[3]==target))
  {
    
    
    ans.add(Arrays.asList(nums[0],nums[1], nums[2], nums[3]));
    return ans;
  }
    Arrays.sort(nums);
    int end=nums.length-1;
     for(int i=0;i<nums.length-3;i++){
    
    
         
          if(i>=1&&nums[i]==nums[i-1])
          continue;
         for(int j=i+1;j<nums.length-2;j++){
    
    
            if(j>i+1&&nums[j]==nums[j-1])
            continue;
         
         int l=j+1;
         int r=end;
         while(l<r){
    
    
             if(nums[i]+nums[l]+nums[r]+nums[j]>target)
             r--;
             else if(nums[i]+nums[l]+nums[r]+nums[j]<target)
             l++;
             else if(nums[i]+nums[l]+nums[r]+nums[j]==target){
    
    

                //  List<Integer>res=new ArrayList<Integer>();
                //  res.add(nums[i]);
                //  res.add(nums[l]);
                //  res.add(nums[r]);
               while (l<r && nums[l] == nums[l+1]) l++; // 去重
               while (l<r && nums[r] == nums[r-1]) r--;     
                  ans.add(Arrays.asList(nums[i],nums[j], nums[l], nums[r]));
                 
                 r--;
                 l++;
             }
         }
         }
     }
     return ans;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_44844588/article/details/108149219
今日推荐