【icyle】Leetcode-cn：18. 四数之和

题目

解答：排序+双指针

思路

同15.三数之和，只是选定一个数加双指针变成了选定两个数加双指针，也就是多了一个for循环。
另外注意剪枝操作即可。

需要的头文件

vector、algorithm

代码

/*
 * @lc app=leetcode.cn id=18 lang=cpp
 *
 * [18] 四数之和
 */

// @lc code=start
#include <vector>
#include <algorithm>
using namespace std;
class Solution
{
public:
    vector<vector<int>> fourSum(vector<int> &nums, int target)
    {
        vector<vector<int>> resnums;
        int numslength = nums.size();
        //特殊情况
        if (numslength < 4)
        {
            return resnums;
        }
        sort(nums.begin(), nums.end());
        //这里排序加双指针法，参考15.三数之和
        //以下是4种减少运行时间的剪枝，来源leetcode官方，讲的很好
        //1.在确定第一个数之后，如果 nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target，说明此时剩下的三个数无论取什么值，四数之和一定大于 target，因此退出第一重循环；
        //2.在确定第一个数之后，如果 nums[i]+nums[n-3]+nums[n-2]+nums[n-1]<nums[i]+nums[n−3]+nums[n−2]+nums[n−1]<target，说明此时剩下的三个数无论取什么值，四数之和一定小于 target，因此第一重循环直接进入下一轮，枚举 nums[i+1]；
        //3.在确定前两个数之后，如果 nums[i]+nums[j]+nums[j+1]+nums[j+2]>nums[i]+nums[j]+nums[j+1]+nums[j+2]>target，说明此时剩下的两个数无论取什么值，四数之和一定大于 target，因此退出第二重循环；
        //4.在确定前两个数之后，如果 nums[i]+nums[j]+nums[n-2]+nums[n-1]<nums[i]+nums[j]+nums[n−2]+nums[n−1]<target，说明此时剩下的两个数无论取什么值，四数之和一定小于 target，因此第二重循环直接进入下一轮，枚举 nums[j+1]。
        //!!!!!注意i的取值：i < numslength - 3，否则一定会发生越界错误！j同理。
        for (int i = 0; i < numslength - 3; i++)
        {
            //注意这里不要漏i>0，因为一开始i=0的话，i-1是非法的
            if (i > 0 && nums[i] == nums[i - 1])
            {
                continue;
            }
            //剪枝1
            if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target)
            {
                return resnums;
            }
            //剪枝2
            if (nums[i] + nums[numslength - 1] + nums[numslength - 2] + nums[numslength - 3] < target)
            {
                continue;
            }
            //进入第二层，也就是第二个数
            for (int j = i + 1; j < numslength - 2; j++)
            {
                if (j > i + 1 && nums[j] == nums[j - 1])
                {
                    continue;
                }
                //剪枝3：注意和剪枝1的区别，这里不能直接退出！
                if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target)
                {
                    break;
                }
                //剪枝4
                if (nums[i] + nums[j] + nums[numslength - 2] + nums[numslength - 1] < target)
                {
                    continue;
                }

                //这里和15.三数之和是一模一样的了，做些许改动即可。
                int left = j + 1, right = numslength - 1;
                while (left < right)
                {
                    int sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum == target)
                    {
                        resnums.push_back({nums[i], nums[j], nums[left], nums[right]});

                        // 这个if循环是找到答案了，那么现在left和right都不需要了，直接移动到下一顺位
                        // 如果出现重复情况，需要再次顺位
                        left++;
                        right--;
                        while (left < right && nums[left] == nums[left - 1])
                            left++;
                        while (left < right && nums[right] == nums[right + 1])
                            right--;
                    }
                    else if (sum < target)
                    {
                        left++;
                    }
                    else if (sum > target)
                    {
                        right--;
                    }
                }
            }
        }
        return resnums;
    }
};
// @lc code=end

时间复杂度和空间复杂度

• 时间复杂度： O(n³) ，原理参考15题，多了一个for循环则对应多一重 n 的时间复杂度。

• 空间复杂度：O(n) 。原理参考15题。

反思与总结