一、题目: 15. 三数之和
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j、i != k 且 j != k ,
同时还满足 nums[i] + nums[j] + nums[k] == 0 。
请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
提示:
- 3 <= nums.length <= 3000
- -105 <= nums[i] <= 105
二、解法
以下解法来自于labuladong的算法小抄
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
// 从小到大排序
Arrays.sort(nums);
// 求符合条件的n元组
return this.nSumTarget(nums, 3, 0, 0);
}
/**
* 求符合条件的n元组
* @param nums 数组 (题目给的)
* @param n 代表求几个数的和
* @param start 数组索引开始位置
* @param target 和的值
* @return 符合条件的n元组
*/
public List<List<Integer>> nSumTarget(int[] nums, int n, int start, int target) {
List<List<Integer>> res = new ArrayList<>();
int size = nums.length;
// 至少应该是2Sum,且数组大小不应该小于n
if (n < 2 || n > size) return res;
// 2Sum是base case
if (n == 2) {
// 双指针
int low = start, high = size - 1;
while (low < high) {
int left = nums[low], right = nums[high];
int sum = left + right;
if (sum < target) {
// 去重(跳过值相同的元素,防止重复)
while (low < high && nums[low] == left) low++;
} else if (sum > target) {
// 去重
while (low < high && nums[high] == right) high--;
} else {
// 两数之和加起来 = 目标值 (sum = target),则保存这两个数
res.add(new ArrayList<>(Arrays.asList(left, right)));
// 去重
while (low < high && nums[low] == left) low++;
while (low < high && nums[high] == right) high--;
}
}
} else {
for (int i = start; i < size; i++) {
// 递归计算后两个数
List<List<Integer>> subList = nSumTarget(nums, n-1, i+1, target-nums[i]);
for (List<Integer> sub : subList) {
// 添加第一个数
sub.add(nums[i]);
res.add(sub);
}
// 去重
while (i < size - 1 && nums[i] == nums[i+1]) i++;
}
}
return res;
}
}
三、思路总结
- 排序 (从小到大)
Arrays.sort(nums);
- 递归计算n元组
- if (n < 2 || n > size)
- 直接return
- else if(n ==2)
- while遍历:从start索引,到末尾
- sum < target
- 去重
- sum > target
- 去重
- sum = target
- 两数添加到列表
- 去重
- else
- for遍历nums
- 递归算出后两个数
- 第一个数添加到列表
- 去重