[LeetCode] 3Sum

最新推荐文章于 2022-01-27 22:44:32 发布

原创最新推荐文章于 2022-01-27 22:44:32 发布 · 463 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #刷题

LeetCode 专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一种解决三数之和问题的有效算法。通过先排序再使用双指针技术，该算法能在O(n^2)的时间复杂度内找到数组中所有相加等于0的唯一三元组组合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

vector<vector<int> > threeSum(vector<int> &num) 
{
	sort(num.begin(), num.end());
	vector<vector<int>> triplets;
	int n = num.size();
	if (n == 0)
	{
		return triplets;
	}
	int lastNum = num[0] + 1;
	for (int aIndex = 0; aIndex < n-2; aIndex++)
	{
		if (num[aIndex] == lastNum)
		{
			continue;
		}
		else
		{
			lastNum = num[aIndex];
		}
		int bIndex = aIndex + 1, cIndex = n-1;
		while (bIndex < cIndex)
		{
			if ( (num[aIndex]+num[bIndex]+num[cIndex]) < 0 )
			{
				bIndex++;
			}
			else if ( (num[aIndex]+num[bIndex]+num[cIndex]) > 0 )
			{
				cIndex--;
			} 
			else
			{
				if (triplets.empty() || (num[bIndex] != triplets.back()[1] || num[cIndex] != triplets.back()[2]))
				{
					vector<int> triplet;
					triplet.push_back(num[aIndex]);
					triplet.push_back(num[bIndex]);
					triplet.push_back(num[cIndex]);
					triplets.push_back(triplet);					
				}
				bIndex++;
				cIndex--;
			}
		}
	}
	return triplets;
}