【Leetcode】18. 4Sum

题目地址：

https://leetcode.com/problems/4sum/

给定一个数组$A$，再给定一个数$t$，要求返回所有$A$里数组成的四元组，使得这个四元组和为$t$。返回的四元组不能有重复。

思路类似于3Sum，先固定前两个数，然后再用双指针的方法做2Sum即可。为了防止重复，可以略过出现多次的数（具体技巧可以参考代码中的写法）。代码如下：

class Solution {
 public:
  vector<vector<int>> fourSum(vector<int>& nums, int target) {
    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 j = i + 1; j < nums.size(); ++j) {
        if (j > i + 1 && nums[j] == nums[j - 1]) continue;
        int l = j + 1, r = nums.size() - 1;
        while (l < r) {
          if (l > j + 1 && nums[l] == nums[l - 1]) {
            ++l;
            continue;
          }
          if (r < nums.size() - 1 && nums[r] == nums[r + 1]) {
            --r;
            continue;
          }
          long long sum = (long long)nums[i] + nums[j] + nums[l] + nums[r];
          if (sum < target)
            ++l;
          else if (sum > target)
            --r;
          else {
            res.push_back({nums[i], nums[j], nums[l], nums[r]});
            ++l;
            --r;
          }
        }
      }
    }

    return res;
  }
};

时间复杂度$O(n^3)$，空间$O(1)$。