浅谈双指针来解决两数之和或三数之和

以力扣的第15题和第18题为例

下面是三数之和：

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

我们正常的方法是三层循环嵌套，但是效率太低了，所以用双指针对上述问题进行优化：

双指针其大体思想是：1.先对数组进行排序，而排序的原因是左右指针其相加大于一个数字，会让指针所指向数字小的往对方的方向走，这就要保证数组的大小的连续性，

2.固定一个数字，使其三数之和变为两数之和

3.接着创建左右指针：

int left = i + 1, right = n - 1, target = -nums[i];

4.两数相加，如果其大于目标数减去固定的一个数字，那么就让左右指针中相对较小的一个向着对方的方向走去；反之，则让较大的向对方走去

5.结束的条件是左指针大于等于右指针，此时将固定的数字向下走一位；

下面给出具体代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        // 1. 排序数组
        sort(nums.begin(), nums.end());
        
        int n = nums.size();
        for (int i = 0; i < n - 2; ++i) {
            // 如果当前数字大于 0，则不存在三数之和为 0 的组合
            if (nums[i] > 0) break;
            // 避免重复计算
            if (i > 0 && nums[i] == nums[i - 1]) continue;
            
            int left = i + 1, right = n - 1, target = -nums[i];
            while (left < right) {
                int sum = nums[left] + nums[right];
                if (sum < target) {
                    ++left;
                } else if (sum > target) {
                    --right;
                } else {
                    // 找到符合条件的组合
                    ret.push_back({nums[i], nums[left], nums[right]});
                    // 避免重复计算
                    while (left < right && nums[left] == nums[left + 1]) ++left;
                    while (left < right && nums[right] == nums[right - 1]) --right;
                    ++left;
                    --right;
                }
            }
        }
        return ret;
    }
};

四数之和，其只需要将三数之和中固定一个数字改为固定两个数字就可以了，下面给出代码：

class Solution {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		vector<vector<int>> ret;
		//排序
		sort(nums.begin(), nums.end());
		//双指针
		int n = nums.size();
		for (int i = 0; i < n;)
		{
			for (int j = i + 1; j < n;)
			{
				int left = j + 1, right = n - 1;
				long long aim = (long long)target - nums[i] - nums[j];
				while (left < right)
				{
					int sum = nums[left] + nums[right];
					if (sum < aim)left++;
					else if (sum > aim)
						right--;
					else
					{
						ret.push_back({ nums[i],nums[j],nums[left++],nums[right--] });
						//去重
						while (left < right&&nums[left]==nums[left-1])
						{
							left++;
						}
						while (left < right && nums[right] == nums[right + 1])
						{
							right--;
						}
					}
				}
				//去重二
				j++;
				while (j < n && nums[j] == nums[j - 1])
				{
					j++;
				}
			}
			//去重三
            i++;
			while (i < n && nums[i] == nums[i - 1])
			{
				i++;
			}
		}
		return ret;
	}
};