LeetCode 18. 四数之和 JAVA

布满荆棘的人生 2022-12-15 14:23 123阅读 0赞

给定一个包含 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]
    ]
    
    来源：力扣（LeetCode）
    链接：https://leetcode-cn.com/problems/4sum
    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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;
                for(int j=i+1;j<nums.length-2;j++)
                {
        
                    if(nums[j]==nums[j-1]&&j!=i+1) continue;
                    int left=j+1;
                    int right=nums.length-1;
                    while(left<right)
                    {
        
                        if(nums[i]+nums[j]+nums[left]+nums[right]<target){
        
                            left++;
                        }
                        else if(nums[i]+nums[j]+nums[left]+nums[right]>target){
        
                            right--;
                        }
                        else{
        
                            List<Integer> tmp=new ArrayList<>();
                            tmp.add(nums[i]);
                            tmp.add(nums[j]);
                            tmp.add(nums[left]);
                            tmp.add(nums[right]);
                            res.add(tmp);
                          //  break;
                            while(left+1<right&&nums[left]==nums[left+1])   
                                left++;
                            left++;
                            while(left<right-1&&nums[right]==nums[right-1])   
                                right--;
                            right--;
                        }
                    }
                }
            }
            return res;
        }
    }