关闭

15. 3Sum

50人阅读 评论(0) 收藏 举报
分类:

先枚举前两个数,再看第三个数在不在,在这里主要需处理好重复的问题,所以先排好序,若第一个数字相同,跳过,若第二个数字想同,也跳过

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n=nums.size();
        int i,j,k,length;
        int ans;
        sort(nums.begin(),nums.end());
        unordered_map<int,int> mp;
        vector<vector<int>> res;
        
        for(i=0;i<n;i++)
            mp[nums[i]]=i;
            
        for(i=0;i<=n-3;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            for(j=i+1;j<n;j++)
            {
                if(j>i+1&&nums[j]==nums[j-1])
                    continue;
                int d=nums[i]+nums[j];
                if(d>0)
                    break;
                if(mp.find(-d)!=mp.end())
                {
                    k=mp[-d];
                    if(k<=j)
                        break;
                    vector<int> vt;
                    vt.push_back(nums[i]);
                    vt.push_back(nums[j]);
                    vt.push_back(nums[k]);
                    res.push_back(vt);
                }
            }
        }
        return res;   
    }
};



另一种方法,不需要空间开销,二分查找,枚举第一个数,第一个数从左端开始找,第二个数从右端开始找

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n=nums.size();
        int i,j,k;
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        
        for(i=0;i<=n-3;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            for(j=i+1,k=n-1;j<k;)
            {
                if(j>i+1&&nums[j]==nums[j-1])
                {
                    j++;
                    continue;
                }
                if(k<n-1&&nums[k]==nums[k+1])
                {
                    k--;
                    continue;
                }
                int d=nums[i]+nums[j]+nums[k];
                if(d==0)
                {
                    vector<int> vt;
                    vt.push_back(nums[i]);
                    vt.push_back(nums[j]);
                    vt.push_back(nums[k]);
                    res.push_back(vt);
                    j++;
                    k--;
                }
                else if(d>0)
                    k--;
                else if(d<0)
                    j++;
            }
        }
        return res;   
    }
};



0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:14044次
    • 积分:1424
    • 等级:
    • 排名:千里之外
    • 原创:135篇
    • 转载:6篇
    • 译文:0篇
    • 评论:0条