给你一个整数数组 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 。
思路:先将数组由小到大排序,然后让从头遍历数组,再设两个左右指针,左指针为从头遍历的指针+1,右指针为数组长度-1;三数之和为0,从头遍历的指针设为i,如果nums[i]>0,就break;
如果nums[i]==nums[i-1] 就continue 然后开始用双指针遍历数组
如果sum<0 就让l++ 跳过重复的相同数字
如果sum>0 就让 r-- 跳过重复的相同数字
sum==0 存入list
public static List<List<Integer>> threeSum(int[] nums) {
ArrayList<List<Integer>> lists = new ArrayList<>();
Arrays.sort(nums);
for (int i = 0; i < nums.length-2; i++) {
int l=i+1;
int r= nums.length-1;
if(nums[i]>0) break;
if(i>0&&nums[i]==nums[i-1]) continue;
while(l<r)
{
int sum = nums[i]+nums[l]+nums[r];
if(sum<0&&l<r)
{
while(l<r&&nums[l]==nums[l++]);
} else if (sum>0&&l<r) {
while(l<r&&nums[r]==nums[r--]);
}else {
ArrayList<Integer> list = new ArrayList<>();
list.add(nums[r]);
list.add(nums[l]);
list.add(nums[i]);
lists.add(list);
while(l<r&&nums[r]==nums[r--]);
while(l<r&&nums[l]==nums[l++]);
}
}
}
return lists;
}