C++~LeetCode18. 四数之和-思路-逐行注释

题目：

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]
 

提示：

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

 来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/4sum

---

题目大致意思很简单：就是从一个数组里面选出四个数作为一个四元组，然后这四个数的和=target，求这样的四元组（四元组不能重复）

思路：

1.首先我们需要排序，方便我们进行去重和遍历操作。

2.然后我们前两个数的和（经过去重的），在剩下的有序区间内，左边界设为left，右边界设为right，分别代表第三第四个数。

3.然后计算四数之和，如果四数之和>target，则让right--（因为数组是有序的，所以让最后一个数缩小，四数之和也就会减小），如果<target，则让left++，如果相等，将四元组加入结果，然后对left和right所指的位置进行去重。（这里需要等到何时的三元组后再去重，直接先去重会漏掉 类似0 0 0 0，target = 0这样的结果）

4.本题最难的是去重，首先需要分别对第一个数和第二个数去重，而且去重的时候必须判断当前元素和前一个元素是否相等，而不是后一个元素，防止漏掉一些情况，比如 1 1 -1 -1 target = 0，往后判断会漏掉这种情况。最后是找到符合条件的四元组后对第三个和第四个数的去重，有重复的就一直缩小区间即可。

代码+注释

class Solution {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		vector<vector<int>> ans;
		sort(nums.begin(), nums.end());  //从小到大排序
		for (int i = 0; i < nums.size(); i++) {
			if (nums[i] > target && nums[i] >= 0) {  //剪枝，如果数组中有元素>target，并且是正数的话，那就一定无结果了，因为我们是从小到大排序过的
				break;
			}
			if (i > 0 && nums[i] == nums[i - 1]) {  //对第一个数进行去重，注意第一个数的去重必须判断当前元素和前一个元素是否相等，防止漏掉一些情况，比如 1 1 -1 -1 target = 0
				continue;
			}
			for (int j = i + 1; j < nums.size(); j++) {
				if (nums[i] + nums[j] > target && nums[i] + nums[j] >= 0) {  //剪枝，此时如果两个数的和 >target且都是整数，直接break
					break;
				}
				if (j > i + 1 && nums[j] == nums[j - 1]) {  //对第二个数进行去重，和第一个数方式一样
					continue;
				}
				int left = j + 1;  //第三个数
				int right = nums.size() - 1;  //第四个数
				while (left < right) {  
					long res = long(nums[i] + nums[j] + nums[left] + nums[right]);  //注意这里要用long强转，否则会溢出
					if (res > target) {  //四数之和 > target，让right--，使得四数之和减小
						right--;
					} else if (res < target) {  //四数之和 < target，让left++，使得四数之和变大
						left++;
					} else {  //四数之和等于target，将四元组加入结果
						vector<int> temp = { nums[i],nums[j],nums[left],nums[right] };
						ans.emplace_back(temp);
						while (left < right && nums[right] == nums[right - 1]) {  //对第四个数去重
							right--;
						}
						while (left < right && nums[left] == nums[left + 1]) {  //对第三个数去重
							left++;
						}
						right--;
						left++;
					}
				}
			}
		}
		return ans;
	}
};

设数组的长度为n，则时间复杂度是O(n^3)，空间复杂度为O(1)