leetcode15 3Sum

最新推荐文章于 2025-09-07 20:32:33 发布

转载最新推荐文章于 2025-09-07 20:32:33 发布 · 61 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://juejin.im/post/5a33b3fe51882529a71def7e

文章标签：

#数据结构与算法

本文介绍了一种算法问题，即在给定的整数数组中找出所有唯一且加和为零的三元组。文章提供了两种解决思路，并详细展示了第二种思路的实现代码，通过先对数组进行排序，然后使用双指针法来高效地解决问题。

题目描述

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is: [ [-1, 0, 1], [-1, -1, 2] ]

思路

给定一个整型数组S，找到所有三个元素加起来为零的数。

思路1.

三个数加起来得是0，那么查找 0 - 前两个数之和是否在数组S里，在就说明S里这三个数相加是0.

那么就插入二维数组中。

排重：新得到的数要和已经在二维数组里的数比较，不相同才可以插入到二维数组。发现超时。

思路2.

首先排序数组S，利用双指针一个在前，一个在后去比较加起来是否为零，然后双指针继续移动。

代码

class Solution {
public:
    vector< vector<int> > threeSum(vector<int>& nums) {
        int i, j, k;
		vector< vector<int> > res;

		if(nums.size() <= 2)
			return res; 

		sort(nums.begin(),nums.end());//排序从小到大

		for(i = 0; i < nums.size() - 2; i++)
		{
			j = i+1;
			k = nums.size() - 1;
			while(j < k)
			{
               	vector<int> initial;
				if(nums[i] + nums[j] + nums[k] == 0)
				{
					initial.push_back(nums[i]);
					initial.push_back(nums[j]);
					initial.push_back(nums[k]);
					res.push_back(initial);
					++j;
					--k;
					while(j < k && nums[j-1] == nums[j])//向后移动了的值与之前相同
					{
						j++;
					}
					while(j < k && nums[k] == nums[k+1])
					{
						k--;
					}
				}
				else if(nums[i] + nums[j] + nums[k] >= 0) //说明太大要缩小
				{
					k--;
				}
				else
					j++;
			}
			while(i < nums.size() - 1 && nums[i] == nums[i+1])
			{
				i++;
			}			
		}
		return res;
    }
};
复制代码