[leetcode 15] 3Sum

三数之和算法解析

最新推荐文章于 2020-06-12 14:31:56 发布

BichonFriseHusky

最新推荐文章于 2020-06-12 14:31:56 发布

阅读量363

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/BichonFriseHusky/article/details/37564341

LeetCode 专栏收录该内容

28 篇文章

订阅专栏

本文探讨了如何找出数组中三个整数相加等于零的所有唯一组合，通过排序和双指针技巧避免重复解，提供了详细的算法实现思路及C++代码。

题目：

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:

Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
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)

思路：

1.此类题目先对数组排序；

2.使第一个数i从头开始遍历数组，另第二个数j=i+1，另第三个数k=nums.size()-1，那么问题转化为在j和k之间的所有数中找到两个数，使得num[j]+num[k]=-num[i]，若两者和小于num[i]，那么j++，若两者和大于num[i]，则k--，若相等，则返回当前i，j，k；

3.题目要求要避免重复，由于数组已排序，那么只要在ijk开始遍历后，若num[j]==num[j-1]（num[k]==num[k+1]），直接continue则可避免结果重复；

4.思路简单，直接看代码；

代码：

class Solution{
public:
	vector<vector<int>> threeSum(vector<int> &num){
		sort(num.begin(),num.end());
		vector<vector<int> > result;
		for(int i=0;i!=num.size();++i)
		{
			if(i>0 && num[i]==num[i-1])
				continue;
			int j=i+1,k=num.size()-1;
			while(j<k)
			{
				if(j>i+1 && num[j]==num[j-1])
				{
					++j;
					continue;
				}
				if(k<(num.size()-1) && num[k]==num[k+1])
				{
					--k;
					continue;
				}
				if(num[i]+num[j]+num[k]<0)
					++j;
				else if(num[i]+num[j]+num[k]>0)
					--k;
				else
				{
					vector<int> t;
					t.push_back(num[i]);
					t.push_back(num[j]);
					t.push_back(num[k]);
					result.push_back(t);	
					++j;--k;
				}
			}
		}
		return result;
	}
};