LeetCode-双指针-15.三数之和

给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。

样例

例1:

输入:[2,7,11,15] 输出:[]

例2:

输入:[-1,0,1,2,-1,-4] 输出:[[-1, 0, 1],[-1, -1, 2]]

注意事项

在三元组(a, b, c),要求a <= b <= c。

vector<vector<int> > threeSum(vector<int> &nums) 
{
vector<vector<int>> res;
//为啥要sort呢,因为要用到双指针来遍历找两数之和,只有在从小到大排序之后的结果上,才能根据判断移动start和end
sort(nums.begin(), nums.end()); 
	for (int k = 0; k < nums.size() - 2; ++k) 
	{
		if (nums[k] > 0) 
			break;
		//处理数组中有相同数的情况
		if (k > 0 && nums[k] == nums[k - 1]) 
			continue;
		//求解两个数相加等于某个数,用双指针法!!!
		int target = 0 - nums[k], i = k + 1, j = nums.size() - 1;
		while (i < j) 
		{
			if (nums[i] + nums[j] == target) 
			{
				res.push_back({nums[k], nums[i], nums[j]});
				//处理数组中有相同数的情况 ,注意,这里内部嵌套的是while语句
				while (i < j && nums[i] == nums[i + 1]) ++i;
				while (i < j && nums[j] == nums[j - 1]) --j;
				++i; --j;
			} 
			// 因为是排序好了的,所以如果 nums[i]+nums[j]<target ,则向右移,反之向左移动
			else if (nums[i] + nums[j] < target) 
				++i;
			else 
				--j;
		}
	}
return res;
}

 

猜你喜欢

转载自blog.csdn.net/qq_32391345/article/details/106845866