LeetCode第15题解析

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

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

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

class Solution {
public:
	vector<vector<int>> threeSum(vector<int>& nums) {
		//存储结果
		vector<vector<int>> result;
		//排序
		sort(nums.begin(), nums.end());
		//考虑边界的情况
		if (nums.size() < 3 || nums.front() > 0 || nums.back() < 0) {
			return result;
		}

		//用k表示固定的数，进行遍历
		for (int k = 0; k < nums.size(); ++k) {
			//如果k递增后值相同，则继续递增
			if (k > 0 && nums[k] == nums[k - 1]) continue;

			//双指针
			int i = k + 1;
			int j = nums.size() - 1;
			while (i < j) {
				//if (nums[k] + nums[i] > 0) break;
				int sum = nums[k] + nums[i] + nums[j];
				if (sum < 0) {
					++i;
				}
				else if (sum > 0) {
					--j;
				}
				else {
					result.push_back({ nums[k], nums[i], nums[j] });
                    ++i;
                    --j;
					while (i < j && nums[i] == nums[i - 1]) ++i;
					while (i < j && nums[j] == nums[j + 1]) --j;
				}
			}
		}
		return result;
	}
};