双指针算法Leetcode018四数之和（基于015三数之和思想）(溢出处理)

地址

https://leetcode-cn.com/problems/4sum/submissions/

描述

思想

三数之和枚举是枚举 i，那么四数之和我们可以枚举 i 和 j 接着使用双指针就行

判断溢出

使用

while(u-1>k&&nums[i]+nums[j]+nums[k]+nums[u-1]>=target) u--;
                    if(nums[i]+nums[j]==target-nums[k]-nums[u]) {
                        res.push_back({nums[i],nums[j],nums[k],nums[u]});
                    }

会导致溢出

解决溢出的办法讲两个数移到等号的右边。

代码

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&&nums[i]==nums[i-1]) continue;
            for(int j=i+1;j<nums.size();j++){
    
    
                if(j>i+1&&nums[j-1]==nums[j]) continue;
                //双指针算法
                for(int k=j+1,u=nums.size()-1;k<u;k++){
    
    
                    if(k>j+1&&nums[k]==nums[k-1]) continue;
                    //试探算法
                    while(u-1>k&&nums[i]+nums[j]>=target-nums[k]-nums[u-1]) u--;
                    if(nums[i]+nums[j]==target-nums[k]-nums[u]) {
    
    
                        res.push_back({
    
    nums[i],nums[j],nums[k],nums[u]});
                    }
                }
            }
        }
        return res;
    }
};