力扣 题目18-- 四数之和

题目

---

题解

---

 

思路:之前做过三数之和 ，那么我们能不能把四数转换成三数呢?

三数

nums[left] + nums[i] + nums[right]==0

四数（这里i我们可以在三数中多加一层循环从左到右 那么此时防止重复i应该从j+1开始 nums[j]也可能重复需要判断）

nums[left] + nums[i]+nums[j] + nums[right]==target

转换一下

nums[left] + nums[i] + nums[right]==target-nums[j]

那么我们改一下条件就能交了

但是我们会发现报错 超过了int的范围

于是我们需要改成

nums[left] + nums[i] ==target-nums[j]-nums[right]

提交

虽然成功了 但是有点不如人意 肯定是在遍历的时候存在一些问题

通过观察可以发现 如果

nums[j] + nums[nums.size() - 1] < target - nums[nums.size() - 3] - nums[nums.size() - 2]

即此轮最大值都小于target 那么就不会其他可能性也不会和target相等 直接跳出即可

同理 此轮最小值都大于target 那么就不会其他可能性也不会和target相等 直接跳出即可

然后我们在两次循环都判断一下这种可能性(别忘记下标出界)即可

可以看见 虽然还是不尽人意但是好多了

代码

---

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> four;
        sort(nums.begin(), nums.end());
        //如果长度小于4 直接返回
        if (nums.size() < 4) {
            return four;
        }
        for (int j = 0; j < nums.size() - 1; j++) {
            //如果nums[j]重复直接跳出
            if (j > 0 && nums[j] == nums[j - 1]) {
                continue;
            }
            //下面同一使用这种比较方式 防止
            //如果最大值都小于target 那么此轮就没有答案跳出
            if (j < (nums.size() - 3) && nums[j] + nums[nums.size() - 1] < target - nums[nums.size() - 3] - nums[nums.size() - 2]) {
                continue;
            }
            //如果最小值都大于target 那么此轮就没有答案跳出
            if (j<(nums.size() - 3) && nums[j] + nums[j + 1] > target - nums[j + 3] - nums[j + 2]) {
                continue;
            }
            for (int i = j + 1; i < nums.size(); i++) {
                //防止nums[i]重复 直接跳出
                if (i > j + 1 && nums[i] == nums[i - 1]) {
                    continue;
                }
                int left = i + 1;
                int right = nums.size() - 1;
                //如果最小值都大于target 那么此轮就没有答案跳出
                if (i < nums.size() - 2 && nums[left] + nums[i]   > target - nums[j] - nums[left + 1]) {
                    break;
                }
                //如果最大值都小于target 那么此轮就没有答案跳出
                if (right > 1 && nums[right - 1] + nums[i] < target - nums[j] - nums[right]) {
                    continue;
                }
                while (left < right)
                {
                    if (nums[left] + nums[i] == target - nums[j] - nums[right]) {
                        four.push_back({ nums[left],nums[i],nums[right],nums[j] });
                        while (left < right && nums[left] == nums[left + 1]) {
                            left = left + 1;
                        }
                        while (left < right && nums[right] == nums[right - 1]) {
                            right = right - 1;
                        }
                        left = left + 1;
                        right = right - 1;
                    }
                    else if (nums[i] + nums[left] > target - nums[j] - nums[right]) {
                        right = right - 1;
                    }
                    else {
                        left = left + 1;
                    }
                }
            }
        }
        return four;
    }
};

int main() {
    Solution sol;
    vector<int> nums = { 1,0,-1,0,-2,2 };
    vector<vector<int>>num = sol.fourSum(nums,0);
    for (int i = 0; i < num.size(); i++) {
        cout << "num[" << i << "]" << "[0]=" << num[i][0] << endl;
        cout << "num[" << i << "]" << "[1]=" << num[i][1] << endl;
        cout << "num[" << i << "]" << "[2]=" << num[i][2] << endl;
        cout << "num[" << i << "]" << "[3]=" << num[i][3] << endl;
    }
}