多指针应用-------三数之和与四数之和

15. 三数之和

难度中等3421

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

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：

输入：nums = []
输出：[]

示例 3：

输入：nums = [0]
输出：[]

class Solution {
public:
    vector<vector<int>> res;

    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        if(nums.size()<3) return {};
        for(int i=0;i<nums.size();i++) {
            while(i>0&&i<nums.size()&&nums[i]==nums[i-1]) {
                i++;
            }
            int left=i+1;
            int right=nums.size()-1;
            // 所以问题就在连续跳过
            while(left<right) {
                int sum = nums[i]+nums[left]+nums[right];
                if(sum==0) {
                    vector<int> result={nums[i],nums[left],nums[right]};
                    res.emplace_back(result);
                    right--;
                     while(left<right&& nums[right]==nums[right+1]) {
                         right--;
                     }
                     left++;
                    while(left<right && nums[left]==nums[left-1]) {
                        left++;
                    }

                }

                if(sum>0) {
                    right--;
                    while(left<right && nums[right]==nums[right+1]) {
                        right--;
                    }
                }

                if(sum<0) {
                    left++;
                    while(left<right && nums[left]==nums[left-1]) {
                        left++;
                    }
                }
            }


        }
            return res;

    }
};

18. 四数之和

难度中等875

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 *a，*b，c 和 d ，使得 a + b + c + 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 = [], target = 0
输出：[]

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

vector<vector<int>> fourSum(vector<int>& nums, int target) {
 // 这个跟三数之和很像
 if (nums.size() < 4) {
   return {};
 }
 sort(nums.begin(), nums.end());
 vector<vector<int>> res;

 for (int i = 0; i < nums.size(); i++) {
   if (i > 0 && nums[i] == nums[i - 1]) {
     continue;
   }
   for (int k = i + 1; k < nums.size() - 2; k++) {  // 优化1
     if (k > i + 1 && nums[k] == nums[k - 1]) {
       continue;
     }
     int l = k + 1;
     int r = nums.size() - 1;
     while (l < r) {
       cout << "i: " << i << " k: " << k << " l: " << l << " r: " << r << endl;
       int sum = nums[i] + nums[k] + nums[l] + nums[r];
       if (sum == target) {
         res.push_back({nums[i], nums[k], nums[l], nums[r]});
         l++;
         r--;
         while (l < r && nums[l] == nums[l - 1]) {
           l++;
         }
         while (l < r && nums[r] == nums[r + 1]) {
           r--;
         }
       } else if (sum < target) {
         l++;
         while (l < r && nums[l] == nums[l - 1]) {
           l++;
         }
       } else if (sum > target) {
         r--;
         while (l < r && nums[r] == nums[r + 1]) {
           r--;
         }
       }
     }
   }
 }

 return res;
}

int main() {
 vector<vector<int>> res;
 vector<int> nums = {0, 0, 0, 0};
 res = fourSum(nums, 1);
 for (auto a : res) {
   for (auto b : a) {
     cout << b << " ";
   }
   cout << endl;
 }
}