Leetcode Problem18

最新推荐文章于 2018-12-26 14:12:14 发布

vandance

最新推荐文章于 2018-12-26 14:12:14 发布

阅读量95

点赞数

分类专栏： leetcode c++ 文章标签： leetcode

本文链接：https://blog.csdn.net/vandance/article/details/81878691

版权

c++ 同时被 2 个专栏收录

48 篇文章 0 订阅

订阅专栏

leetcode

43 篇文章 0 订阅

订阅专栏

4Sum

问题描述：给定一个数组，要求一个四元组的四个数之和为target，找出所有不重复的四元组。

这道题跟之前的3Sum是同一类型的题，就是在3Sum的基础上再加上一个循环。

vector<vector<int>> fourSum(vector<int>& nums,int target)
{
    vector<vector<int>> res;
    if(nums.size()<4) return res;
    sort(nums.begin(),nums.end());
    for(int i=0;i<nums.size()-3;i++)
    {
        if(i>0&&nums[i]==nums[i-1]) continue; //去除重复
        for(int j=i+1;j<nums.size()-2;j++)
        {
            if(j>i+1&&nums[j]==nums[j-1]) continue; //去除重复
            int m=j+1,n=nums.size()-1;
            while(m<n)
            {
                if(nums[i]+nums[j]+nums[m]+nums[n]==target)
                {
                    res.push_back({nums[i],nums[j],nums[m],nums[n]});
                    m++;n--;
                    while(m<n&&nums[m]==nums[m-1]) m++; //去除重复
                    while(m<n&&nums[n]==nums[n+1]) n--; //去除重复
                }
                else if(nums[i]+nums[j]+nums[m]+nums[n]<target)
                {
                    m++;
                    while(m<n&&nums[m]==nums[m-1]) m++;
                }   
                else
                {
                    n--;    
                    while(m<n&&nums[n]==nums[n+1]) n--;
                }

            }
        }       
    }
    return res;
}