题目:
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。
请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
C++:
双指针法:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;class Solution {
public:vector<vector<int>> threeSum(vector<int>& nums) {int n = nums.size();sort(nums.begin(), nums.end()); // 对数组进行排序,以便后续操作vector<vector<int>> answer; // 存储结果的二维向量for (int i = 0; i < n; i++) { // 遍历数组,固定第一个元素// 避免重复的固定元素if (i > 0 && nums[i] == nums[i - 1])continue;int left = i + 1; // 左指针指向固定元素的下一位int right = n - 1; // 右指针指向数组末尾while (left < right) {int sum = nums[i] + nums[left] + nums[right]; // 计算三个元素的和if (sum < 0) { // 如果和小于零,说明需要增大和,左指针右移一位left++;}else if (sum > 0) { // 如果和大于零,说明需要减小和,右指针左移一位right--;}else { // 和等于零,找到满足条件的三元组answer.push_back(vector<int>{ nums[i], nums[left], nums[right] }); // 将三元组添加到结果中cout << 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 answer;}
};int main()
{Solution S;vector<int> nums = { -1, 0, 1, 2, -1, -4 };vector<vector<int>> answers = S.threeSum(nums);
}
python:
思路:先排序,然后两边向中间靠拢。
class Solution():def threeSum(self,nums):nums.sort() # 排序res = []for i in range(len(nums)): # 遍历每一个数if i==0 or nums[i] > nums[i-1]: # 确定不重复的数字(开头)l = i+1r = len(nums)-1while l<r:s = nums[i] + nums[l] + nums[r]if s == 0:res.append([nums[i],nums[l],nums[r]])l += 1r -= 1# 左边向右移动到不重复数为止while l<r and nums[l] == nums[l-1] :l += 1# 右边向左边移动不重复数为止while l<r and nums[r] == nums[r+1] :r -= 1elif s >0:r -= 1else:l += 1return res