Leetcode18. 四数之和

18. 四数之和

Difficulty: 中等

给定一个包含 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]
]

Solution

Language: java

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        if(nums.length < 4) return res;
        for(int i=0; i<nums.length - 1; i++){
            if(i!=0 && nums[i-1]==nums[i]) continue;
            for(int j=i+1; j<nums.length; j++){
                if(j!=i+1 && nums[j-1] == nums[j]) continue;
                int start = j + 1, end = nums.length-1;
                while(start < end){
                    int sum = nums[start] + nums[end] + nums[i] + nums[j];
                    if(sum == target){
                        List<Integer> t = new ArrayList<>();
                        t.add(nums[start]);
                        t.add(nums[end]);
                        t.add(nums[i]);
                        t.add(nums[j]);
                        res.add(t);
                        while(start < end && nums[start] == nums[++start]) ;
                        while(start < end && nums[end] == nums[--end]) ;
                    }else if(sum < target){
                        while(start < end && nums[start] == nums[++start]) ;
                    }else{
                        while(start < end && nums[end] == nums[--end]) ;
                    }
                }
            }
        }
        return res;
    }
}
​