Leecode热题100---15:三数之和为零

题目:

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。

请你返回所有和为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

C++:

双指针法:

cpp 复制代码
#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:

思路:先排序,然后两边向中间靠拢。

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+1
                r = len(nums)-1
                while l<r:
                    s = nums[i] + nums[l] + nums[r]
                    if s == 0:
                        res.append([nums[i],nums[l],nums[r]])
                        l += 1
                        r -= 1
                        # 左边向右移动到不重复数为止
                        while l<r and nums[l] == nums[l-1] :
                            l += 1
                        # 右边向左边移动不重复数为止
                        while l<r and nums[r] == nums[r+1] :
                            r -= 1
                    elif s >0:
                        r -= 1
                    else:
                        l += 1
        return res
相关推荐
此生只爱蛋17 分钟前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
咕咕吖1 小时前
对称二叉树(力扣101)
算法·leetcode·职场和发展
九圣残炎1 小时前
【从零开始的LeetCode-算法】1456. 定长子串中元音的最大数目
java·算法·leetcode
lulu_gh_yu1 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法
丫头,冲鸭!!!2 小时前
B树(B-Tree)和B+树(B+ Tree)
笔记·算法
Re.不晚2 小时前
Java入门15——抽象类
java·开发语言·学习·算法·intellij-idea
为什么这亚子3 小时前
九、Go语言快速入门之map
运维·开发语言·后端·算法·云原生·golang·云计算
3 小时前
开源竞争-数据驱动成长-11/05-大专生的思考
人工智能·笔记·学习·算法·机器学习
~yY…s<#>3 小时前
【刷题17】最小栈、栈的压入弹出、逆波兰表达式
c语言·数据结构·c++·算法·leetcode
幸运超级加倍~4 小时前
软件设计师-上午题-16 算法(4-5分)
笔记·算法