给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
首先求满足条件的两个数,然后将三个数之和转变为求两数之和。那么,确定了第一个数字之后,剩下的两个数字可以是什么呢?其实就是和为 target - nums[i] 的两个数字呗,那不就是 twoSum 函数解决的问题么
class Solution {
public List<List<Integer>> twoSum(int[] nums, int start, int target) {
// 创建一个二维数组,存放满足条件的结果
List<List<Integer>> res = new ArrayList<>();
//左右指针
int lo=start, hi = nums.length-1;
while (lo<hi){
int sum = nums[lo] + nums[hi];
//记录左右指针的初始值
int left = nums[lo], right = nums[hi];
//根据sum和target的值以此移动指针
if(sum<target) {
while (lo<hi && nums[lo] == left) lo++;
}
if(sum>target){
while (lo<hi && nums[hi] == right) hi--;
}
if(sum==target){
//记录满足要求的值//
res.add(new ArrayList<Integer>(Arrays.asList(nums[lo], nums[hi])));
//跳过重复值的元素
while (lo<hi && nums[lo] == left) lo++;
while (lo<hi && nums[hi] == right) hi--;
}
}
return res;
}
public List<List<Integer>> threeSum(int[] nums) {
// target = 0
int target = 0;
//对数组进行排序
Arrays.sort(nums);
// 定义一个结果集
List<List<Integer>> res = new ArrayList<>();
List<List<Integer>> tempres = new ArrayList<>();
for(int i=0;i<nums.length;i++){
System.out.println(i);
// 计算target-nums[i]
tempres = twoSum(nums,i+1,target-nums[i]);
//将返回的的结果,加上i就是三元组的结果
// 将三数的结果集加入到结果集中
if(tempres!=null){
for (int j=0;j<tempres.size();j++){
res.add(Arrays.asList(nums[i], tempres.get(j).get(0),tempres.get(j).get(1)));
System.out.println(j);
}
}
//跳过第一个值重复的情况
while (i<nums.length-1 && nums[i] == nums[i+1]) i++;
}
return res;
}
}