给你一个整数数组 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 。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
Arrays.sort(nums);
for(int i = 0;i < nums.length;i++){
if(nums[i] > 0){
return res;
}
if(i > 0 && nums[i] == nums[i-1]){
continue;
}
int l = i + 1, r = nums.length - 1;
while(l < r){
int sum = nums[i] + nums[l] + nums[r];
if(sum > 0){
r--;
}else if(sum < 0){
l++;
}else{
res.add(Arrays.asList(nums[i] , nums[l] , nums[r]));
while(l < r && nums[r] == nums[r-1]) r--;
while(l < r && nums[l] == nums[l+1]) l++;
r--;
l++;
}
-
首先,创建一个空的结果列表
res
,用于存储找到的三元组。 -
对数组
nums
进行排序,这样可以更容易地处理重复元素,并且有利于使用双指针进行查找。 -
开始遍历数组
nums
中的每个元素,以当前元素作为第一个数字nums[i]
:- 如果当前元素
nums[i]
大于 0,则表示后面的元素都大于 0,因为数组已经排序,所以直接返回结果列表res
。 - 如果当前元素
nums[i]
等于前一个元素nums[i-1]
,则说明已经处理过这个元素,跳过这次循环。
- 如果当前元素
-
在当前元素的右侧区间内,使用双指针
l
和r
分别指向当前元素的下一个元素和数组末尾。 -
在内部的
while
循环中,不断移动指针l
和r
,直到它们相遇为止:- 计算当前三个元素的和
sum = nums[i] + nums[l] + nums[r]
。 - 如果
sum > 0
,则右指针左移一位,减小右侧元素的值,以使总和减小。 - 如果
sum < 0
,则左指针右移一位,增大左侧元素的值,以使总和增加。 - 如果
sum == 0
,则将当前三元组[nums[i], nums[l], nums[r]]
添加到结果列表res
中,并进一步排除重复解,即跳过所有与当前解相同的连续元素。
- 计算当前三个元素的和
-
最后返回结果列表
res
,其中包含了所有和为 0 的三元组。