题目描述
给定一个包含 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]
输出： []
提示：
0 <= nums.length <= 3000-105 <= nums[i] <= 105
注意：本题与主站 15 题相同：leetcode-cn.com/problems/3s…
解法
枚举第一个数，然后用双指针确定另外两个数
Python3
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n, ans = len(nums), []
nums.sort()
for i in range(n - 2):
if i > 0 and nums[i] == nums[i - 1]: continue
left, right = i + 1, n - 1
while left < right:
cur = nums[i] + nums[left] + nums[right]
if cur < 0:
left += 1
elif cur > 0:
right -= 1
else:
ans.append([nums[i], nums[left], nums[right]])
while left < right and nums[left] == nums[left + 1]:
left += 1
while left < right and nums[right] == nums[right - 1]:
right -= 1
left += 1
right -= 1
return ans
Java
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int n = nums.length;
List<List<Integer>> ans = new ArrayList<>();
Arrays.sort(nums);
for (int i = 0; i < n - 2; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1, right = n - 1;
while (left < right) {
int cur = nums[i] + nums[left] + nums[right];
if (cur < 0) {
left++;
} else if (cur > 0) {
right--;
} else {
ans.add(Arrays.asList(nums[i], nums[left], nums[right]));
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
left++;
right--;
}
}
}
return ans;
}
}
