题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路:排序+双指针
题目中要求说是不重复的三元组,那么不重复的本质又是什么呢?
在保持三重循环的大框架不变的情况下,我们只需要保证:
(1) 第二重循环枚举到的元素不小于当前第一重村换枚举到的元素
(2) 第三重循环枚举到的元素不小于当前第二重循环枚举到的元素。
要实现这一点,可以将数组中的元素从小到大进行排序,随后使用普通的三重循环就可以满足上面的要求。
同时,对于每一重循环而言,相邻两个枚举的元素不能相同,否则也会造成重复。
但是三重循环时间复杂度过高,我们可以继续进行优化。如果我们固定了前两重循环枚举到的元素a和b,由于a+b+c=0,则只有唯一的c满足条件。当第二重循环往后枚举一个元素b’时,由于b’ > b,那么满足 a+b’+c’=0的c’一定有 c’ < c,即 c’ 在数组中一定出现在 c 的左侧。也就是说,可以从小到大枚举b,同时从大到小枚举c,即第二重循环和第三重循环实际上是并列的关系。
基于以上思想,我们可以保持第二重循环不变,将第三重循环变成一个从数组最右端开始向左移动的指针。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int len = nums.length;
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<>();
//枚举a
for(int first = 0; first < len; first++){
//需要和上一次枚举到的值不相同
if(first > 0 && nums[first] == nums[first-1]){
continue;
}
//c对应的指针初始指向数组的最右端
int third = len - 1;
int target = -nums[first];//接下来就是寻找第二个和第三个数
//枚举b
for(int second = first + 1; second < len; second++){
//需要和上次枚举到的值不相同
if(second > first+1 && nums[second] == nums[second-1]){
continue;
}
//需要保证b的指针在c指针的左侧
while(second < third && nums[second] + nums[third] > target){
third--;//第三个指针前移
}
//如果指针重合,随着b后续的增加就不会有满足a+b+c=0且b<c的c了,退出循环
if(second == third){
break;
}
if(nums[second] + nums[third] == target){
List<Integer> list = new ArrayList<>();
list.add(nums[first]);
list.add(nums[second]);
list.add(nums[third]);
ans.add(list);
}
}
}
return ans;
}
}