算法训练营第十四天 | 18. 四数之和

给你一个由n个整数组成的数组nums，和一个目标值target。请你找出并返回满足下述全部条件且不重复的四元组[nums[a], nums[b], nums[c], nums[d]]（若两个四元组元素一一对应，则认为两个四元组重复）：

• 0 <= a, b, c, d < n
• a、b、c和d互不相同
• nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按任意顺序返回答案 。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8输出：[[2,2,2,2]]

提示：

• 1 <= nums.length <= 200
• -109 <= nums[i] <= 109
• -109 <= target <= 109

public List<List<Integer>> fourSum(int[] nums, int target) { Arrays.sort(nums); return nSumTarget(nums, 4, 0, (long) target); } public List<List<Integer>> nSumTarget(int[] nums, int n, int start, long target) { List<List<Integer>> res = new ArrayList<>(); int numsLength = nums.length; //n不能小于2，或者数组长度不能小于n if (n < 2 || numsLength < n) { return res; } //两数和的问题，使用双指针 if (n == 2) { int left = start, right = numsLength - 1; while (left < right) { int num1 = nums[left], num2 = nums[right]; long sum = num1 + num2; if (sum == target) { res.add(new ArrayList<>(Arrays.asList(num1, num2))); while (left < right && num1 == nums[left]) { left++; } while (left < right && num2 == nums[right]) { right--; } } else if (sum < target) { while (left < right && num1 == nums[left]) { left++; } } else { while (left < right && num2 == nums[right]) { right--; } } } } else { //n数之和的问题，递归处理 for (int i = start; i < numsLength; i++) { int baseNum = nums[i]; List<List<Integer>> nSumTarget = nSumTarget(nums, n - 1, i + 1, target - baseNum); for (List<Integer> list : nSumTarget) { list.add(baseNum); res.add(list); } while (i < numsLength - 1 && nums[i] == nums[i + 1]) { i++; } } } return res; }