题目:
给你一个整数数组 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] 。
注意,输出的顺序和三元组的顺序并不重要。
思路解析
**1.**题目中要求找到所有「不重复」且和为 000 的三元组。所以先进行一次排序。
2.可以发现,如果我们固定了前两重循环枚举到的元素 a和 b,那么只有唯一的 c 满足 a+b+c=0。当第二重循环往后枚举一个元素 b2时,由于 b2>b,那么满足 a+b2+c2=0 的 c2一定有 c2<c ,即c2在数组中一定出现在 c 的左侧。也就是说,我们可以从小到大枚举 b,同时从大到小枚举 c,即第二重循环和第三重循环实际上是可以合并的
就是双指针
3.完成上述步骤后,可以继续优化,比如减少循环中相同数字的重复查询
下面是完整代码,注释有写
java
public List<List<Integer>> threeSum(int[] nums) {
int n =nums.length;
Arrays.sort(nums);
List<List<Integer>> answer =new ArrayList<>();
for (int first =0;first <n;first++){
// 需要和上一次枚举的数不相同
if (first >0 && nums[first] == nums[first -1]){
continue;
}
// c 对应的指针初始指向数组的最右端
int cright =n-1;
int target = -nums[first];// b+c 的值
for(int second =first+1; second <n;second++){
// 需要和上一次枚举的数不相同
if (second >first+1 && nums[second] == nums[second -1]){
continue;
}
// 需要保证 b 的指针在 c 的指针的左侧
while (second <cright && nums[second]+nums[cright]>target){
cright--;
}
// 如果指针重合,随着 b 后续的增加
// 就不会有满足 a+b+c=0 并且 b<c 的 c 了,可以退出循环
if (second == cright) {
break;
}
if(nums[second]+nums[cright] == target){
List<Integer> list = new ArrayList<Integer>();
list.add(nums[first]);
list.add(nums[second]);
list.add(nums[cright]);
answer.add(list);
}
}
}
return answer;
}