4 Sum Problem

Table of contents

No heading

No headings in the article.


Given an array nums of n integers, return an array of all the unique quadruplets [nums[a], nums[b], nums[c], nums[d]] such that:

  • 0 <= a, b, c, d < n

  • a, b, c, and d are distinct.

  • nums[a] + nums[b] + nums[c] + nums[d] == target

You may return the answer in any order.

Example 1:

Input: nums = [1,0,-1,0,-2,2], target = 0
Output: [[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

Example 2:

Input: nums = [2,2,2,2,2], target = 8
Output: [[2,2,2,2]]

Constraints:

  • 1 <= nums.length <= 200

  • -109 <= nums[i] <= 109

  • -109 <= target <= 109

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        long targe = target;
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<>();
        int n = nums.length;
        if (n < 4) return ans;
        for(int i=0;i<n-3;i++){
            for(int j=i+1;j<n-2;j++){
                int left = j+1,right = n-1;
                long tar = targe - nums[i] - nums[j];
                while(left < right){
                    long cur = nums[left] + nums[right];
                    if(cur == tar){
                        List<Integer> quad = new ArrayList<>();
                        quad.add(nums[i]);
                        quad.add(nums[j]);
                        quad.add(nums[left]);
                        quad.add(nums[right]);
                        ans.add(quad);
                        while(++left < n && nums[left] == quad.get(2));
                        while(--right > 0 && nums[right] == quad.get(3));
                    }else if(cur > tar) right--;
                    else left++;
                }
                while(j+1 < n-2 && nums[j] == nums[j+1]) j++;
            }
            while(i+1 < n-3 && nums[i] == nums[i+1]) i++;
        }
        return ans;

    }
}