题目
给你一个包含 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]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
思路:三重循环
类似于题目两数之和(给定数组和目标数值,求和等于目标数值的两个数)
这一题可以把一个数a当作target,另两个数(b,c)为加数,转换为两数之和。
难点:不重复的三元组
为了不重复
- 需要把数组排序,a指针指向最小值(首),然后在a之后的数字中枚举b,c
当b,c重合时结束枚举,当a+b+c<=0时结束枚举,并检查是否a+b+c=0 - 需要适当地去除重复的数字,在经过一次之后,就跳过后面的数字。所以需要对指针前面的数字比较,此处需要注意越界问题,在首次枚举时不需要检查,c指针不需要检查,因为b指针会检查。因为b,c的数字可能会相同,检查c的指针会帮b跳过数字。
JAVA
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
int n=nums.length;
for(int font=0;font<n;++font){
//去除重复
if(font>0&&nums[font]==nums[font-1])
continue;
for(int mid=font+1;mid<n;++mid){
//去除重复
if(mid>font+1&&nums[mid]==nums[mid-1]){
continue;
}
int rear=n-1;
for(rear=n-1;rear>0;--rear){
//在大于等于时退出
if(nums[rear]+nums[mid]+nums[font]<=0)
break;
}
//这些判断部分如果放在第三重循环,会超出时间限制
if(rear<=mid)break;//先判断
if(nums[rear]+nums[mid]+nums[font]==0){
List list = new ArrayList();
list.add(nums[font]);
list.add(nums[mid]);
list.add(nums[rear]);
res.add(list);
}
}
}
return res;