https://leetcode.cn/problems/4sum/description/

<aside> 💡

思路与三数之和相同,不过变为固定两个数。和 454. 四数相加 II 相比难点在去重上。

细节有点多,尤其注意目标值变为了一个可变的target,target可能为负,所以剪枝的条件与三数之和有所不同。

</aside>

vector<vector<int>> fourSum(vector<int>& nums, int target) {
    vector<vector<int>> ans;
    std::sort(nums.begin(), nums.end());

    // 固定i与j,移动left和right
    for (int i = 0; i < nums.size(); ++i) {
        // 根据nums[i]的取值进行剪枝
        if (nums[i] > target && nums[i] >= 0) {
            break;
        }
        // 对i去重
        if (i > 0 && nums[i] == nums[i - 1]) {
            continue;
        }

        for (int j = i + 1; j < nums.size(); ++j) {
            // 根据nums[i]与nums[j]的取值进行剪枝
            if (nums[i] + nums[j] > target && nums[j] >= 0) {
                break;
            }
            // 对j去重
            if (j > i + 1 && nums[j] == nums[j - 1]) {
                continue;
            }

            int left = j + 1;
            int right = nums.size() - 1;
            while (left < right) {
                long sum = nums[i] + nums[j] + nums[left] + nums[right];
                if (sum > target) {
                    --right;
                } else if(sum < target){
                    ++left;
                } else {
                    ans.push_back({ nums[i], nums[j], nums[left], nums[right] });
                    while (left < right && nums[left + 1] == nums[left]) {
                        ++left;
                    }
                    while (left < right && nums[right - 1] == nums[right]) {
                        --right;
                    }
                    ++left;
                    --right;
                }
            }
        }
    }
    return ans;
}