leetcode三数之和及相似题型（c++）

三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

数组问题大多课用双指针法解决，这题也不例外。这类型题都是先排序后比较容易做。接着先固定一个点，在用双指针夹逼遍历数组来得出结果。只要注意过滤掉重复问题就可以了，即下一个与上一个相等时跳过。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) 
    {
        vector<vector<int>> ans;//保存结果
        int len=nums.size();//数组长度
        if(len<3)//小于3个直接退出
            return ans; 
        sort(nums.begin(),nums.end());//排序
        for(int i=0;i<len-2;++i)//i最多到倒数第三个
        {
            int l=i+1;//i后面一个
            int r=len-1;//数组最后一个
            while(l<r)//固定i，移动l或r遍历数组
            {
                int sum=nums[i]+nums[l]+nums[r];//计算三个数和
                if(sum==0)//等于0即找到
                {
                    ans.push_back(vector<int>{nums[i],nums[l],nums[r]});//将三个数放进输出结果中
                    while(l+1<r&&nums[l]==nums[l+1])//l下一个与当前相同跳过
                        ++l;
                    while(r-1>l&&nums[r]==nums[r-1])//r下一个与当前相同跳过
                        --r;
                    ++l;//找到后，l和r同时往里走一步
                    --r;
                }
                else if(sum<0)//因为数组已经排好序，sum<0,移动l，sum增大
                {
                    ++l;
                }
                else//与小于0相反
                    --r;
            }
            while(i+1<len-2&&nums[i+1]==nums[i])//i的下一个与前一个相同跳过
                ++i;
        }
        return ans;
    }
};

最接近的三数之和

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数，使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

思路

与上一题差不多的类型，只不过是换成了最相近的数，整体思路还是一致，知识多用了一个变量来存放当前最接近的值。

代码

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) 
    {
        int len=nums.size();//获取数组长度
        if(len<3)//长度小于3，返回一个异常值
            return INT_MIN;
        
        sort(nums.begin(),nums.end());//排序
        int sum=nums[0]+nums[1]+nums[2];//设置一个初值，或者设置一个必然取不到的值，但还是取一个存在的值比较稳妥
        if(sum==target)//若该值与目标值一致则最小，返回
            return target;
        for(int i=0;i<len-2;++i)
        {
            int l=i+1;//定义左右指针
            int r=len-1;
            while(l<r)
            {
                int temp=nums[i]+nums[l]+nums[r];//计算当前总和
                sum=abs(temp-target)<abs(sum-target)?temp:sum;//当前总和和上一个最接近数哪个比较接近目标值，记录最接近的值
                if(sum==target)//相同，返回
                    return target;  
                if(temp<target)//nums有序，小于target，往里移动l，temp增大
                    ++l;
                else if(temp>target)//大于target，往里移动r，temp减小
                    --r;
            }
        }
        return sum;
    }
};

四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。
注意：
答案中不可以包含重复的四元组。
示例：
给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。
满足要求的四元组集合为：
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

思路

也是同一类问题，只不过三数之和是一个定点计算，四数之和用两个定点计算，因此需要多用一个循环来迭代，除此之外，其他操作也与三数之和大同小异。

代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) 
    {
        int len=nums.size();//获取数组长度
        vector<vector<int>> ans;//保存结果
        if(len<4)//小于4个，返回空
            return ans;
        
        sort(nums.begin(),nums.end());//排序
        
        for(int i=0;i<len-3;++i)//迭代第一个定点
        {
            for(int j=i+1;j<len-2;++j)//迭代第二个定点
            {
                int l=j+1;//定义左右指针
                int r=len-1;
                
                while(l<r)
                {
                    int sum=nums[i]+nums[j]+nums[l]+nums[r];//计算当前四数之和
                    if(sum==target)//相等，记录这四个数
                    {
                        ans.push_back(vector<int>{nums[i],nums[j],nums[l],nums[r]});
                        while(l+1<r&&nums[l+1]==nums[l])//过滤相等值
                            ++l;
                        while(r-1>l&&nums[r-1]==nums[r])
                            --r;
                        ++l;
                        --r;
                    }
                    else if(sum<target)//与三数之和一致
                        ++l;
                    else
                        --r;
                }
                while(j+1<len-2&&nums[j+1]==nums[j])//过滤第二个相等的定点
                    ++j;
            }
            while(i+1<len-3&&nums[i+1]==nums[i])//过滤第一个相等的定点
                ++i;
        }
        return ans;
    }
};