LeetCode—15.三数之和(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]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 双指针法

（1）思路

通过排序使数组变为有序，便于后续指针移动求解。
先固定第1个数，再通过双指针不断缩小第2、3个数的求解范围。
注意点：
1）nums 中个数小于3时的边界情况；
2）双指针移动过程中对重复数据的处理；
3）第1个数为最小值，因此第1个数大于0时可跳出循环。
时间复杂度：O(nlogn) + O(n * n) = O(n ^ 2)

（2）代码

class Solution {
public:
	vector<vector<int>> threeSum(vector<int>& nums) {
		vector<vector<int>> ans;
		int nsize = nums.size();
		if (nsize < 3)
			return ans;
		sort(nums.begin(), nums.end());
		for (int i = 0; i < nsize && nums[i] <= 0; i++) {
			if (i && nums[i - 1] == nums[i])
				continue;
			int l = i + 1, r = nums.size() - 1;
			int target = 0 - nums[i];
			while (l < r) {
				if (nums[l] + nums[r] < target)
					l++;
				else if (nums[l] + nums[r] > target)
					r--;
				else {
					vector<int> ians = { nums[i], nums[l], nums[r] };
					ans.push_back(ians);
					l++;
					r--;
					while (l < r && nums[l - 1] == nums[l])
						l++;
					while (l < r && nums[r + 1] == nums[r])
						r--;
				}
			}
		}
		return ans;
	}
};

（3）结果

执行用时 ：116 ms, 在所有 C++ 提交中击败了97.81%的用户；
内存消耗 ：15.8 MB, 在所有 C++ 提交中击败了78.97%的用户。

三、其他

暂无。