题目:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
直接看题干,使得a + b + c = 0,三个变量,我们大概可以用三个for循环暴力求解,但是O(n^3)的复杂度太大,肯定会超时。这里我们想一下,如果是a + b = 0两数之和,我们很容易想到先排序然后再使用双指针,这里我们将a+b+c =0变式为a+b=-c,这样我们只是在两数之和的基础上再加一个for循环枚举。
代码如下:
class Solution {
public <List> threeSum(int[] nums) {
int n = nums.length;
Arrays.sort(nums);
List> ans = new ArrayList>();
// 枚举 a
for (int first = 0; first < n; ++first) { // 需要和上一次枚举的数不相同 if (first > 0 && nums[first] == nums[first - 1])
{
continue;
}
// c 对应的指针初始指向数组的最右端
int third = n - 1;
int target = -nums[first];
// 枚举 b
for (int second = first + 1; second < n; ++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<Integer>();
list.add(nums[first]);
list.add(nums[second]);
list.add(nums[third]);
ans.add(list);
}
}
}
return ans;
}
}