@LeetCode三数之和--3Sum[C++]

问题描述

给定一个包含 n 个整数的数组 nums，判断nums是否存在三个元素a, b, c，使得 $a+b+c=0$ ？找出所有满足条件且不重复的三元组。

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

例如，给定数组 nums = [-1, 0, 1, 2, -1, -4].
满足要求的三元组集合为：
[-1, 0, 1]
[-1, -1, 2]

解决方法及复杂度分析

这类问题从 2 数之和到 4 数之和，之后我会分享 4 数之和如何解决，与 3 数之和方法大同小异。

我们引入 3 个指针，分别指向数组中三个元素。为了方便遍历数组，避免出现重复遍历的情况，首先将输入数组进行排序。接下来，前 2 个指针从前向后遍历，剩下的指针从后向前遍历，如果三数之和大于零，就前移第 3 个指针；否则，后移第 2 个指针。

为了保证三元组不重复，我们采取如下方式：如果指针指向的下一个元素与当前指向元素相同，则指针移动到下一个元素。

程序实现

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

@北京·怀柔 2019.3.2