LeetCode-18-4Sum(四数之和)

题目

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

注意：

答案中不可以包含重复的四元组。

示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum

解答

解题思路完全和三数之和一致，因为用了三重循环，所以速度不会太快。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
       Arrays.sort(nums);
        List<List<Integer>> re = new ArrayList<>();
        if (nums.length < 4) {
			return re;
		}
        
        for(int i = 0; i < nums.length; i ++) {
        	if (i > 0 && nums[i] == nums[i - 1]) {
				continue;
			}
        	for(int j = i + 1; j < nums.length; j++) {
        		if (j > i + 1 && nums[j] == nums[j - 1]) {
					continue;
				}
        		int pre = j + 1;
        		int last = nums.length - 1;
        		
        		while(pre < last) {
        			int temp = nums[i] + nums[j] + nums[pre] + nums[last];
        			
        			if (temp == target) {
						List<Integer> list = new ArrayList<>();
						list.add(nums[i]);
						list.add(nums[j]);
						list.add(nums[pre]);
						list.add(nums[last]);
						re.add(list);
						
						while(pre < last && nums[pre] == nums[pre + 1]) pre ++;
						while(pre < last && nums[last] == nums[last - 1]) last --;
						pre ++;
						last --;
					}else if (temp > target) {
						last --;
					}else {
						pre ++;
					}
        		}
        	}
        }
        return re;
    }
}

结果