3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.
For example, given array S = {-1 0 1 2 -1 -4},

A solution set is:
(-1, 0, 1)
(-1, -1, 2)

思路:第一眼看到这题目想到了2Sum,也就是使用哈系表来降低时间复杂度,这道题使用暴力解决(3重循环)的话时间复杂度是O(n3),使用哈系表能够使得时间复杂度降为O(n2),具体做法是,-a=b+c,将-a当作target,剩下的就是2sum的问题了,一个技巧是,顺序将nums中的元素作为target时,求解2Sum只需从target之后的元素中求解即可,同时需要注意可满足解的重复问题。

class Solution {
public:

    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int> > res;
        res.clear();
        int len = nums.size();
        sort(nums.begin(), nums.end());
        for(int i = 0; i != len;++i){
            if(i > 0 && nums[i]==nums[i-1])//相同的target,子问题已求解
                continue;
            //2Sum问题
            map<int, int>hmap;
            for(int j = i+1; j < len; ++j){
                //去除重复
                if(j > i+1 && nums[j] == nums[j-1] ){//从i+1开始连续两个数相同
                    if(nums[i] + nums[j] + nums[j-1])//如果不能构成可满足解,即该子问题已经求解(例如-211与-311的区别)
                        continue;
                    if(j-2 > i &&nums[j] == nums[j-2])//如果在i之后有连续三个相同的数,例如0000,该子问题同样求解
                        continue;
                }
                if(!hmap.count(nums[j]))
                    hmap.insert(pair<int,int>(nums[j],j));
                if(hmap.count(-nums[i]-nums[j])){
                    int n = hmap[-nums[i]-nums[j]];
                    if(n < j){
                        vector<int> tmp;
                        tmp.push_back(nums[i]);
                        tmp.push_back(nums[j]);
                        tmp.push_back(-nums[i]-nums[j]);
                        sort(tmp.begin(),tmp.end());
                        res.push_back(tmp);
                    }
                }
            }
        }
        return res;
    }
};

通过查看相关的资料后,发现还有时间复杂度为O(n2)的解法:

  1. 首先将数组排序
  2. 遍历数组,假设当前a指向nums[i],b指向nums[i1],c指向nums[size-1]现在a+b+c有以下三种情况:
    (1)>0,说明c偏大,c–
    (2)<0,说明b偏小,b++
    (3)=0,找到一个可满足的解
    需要注意的是要解决可满足解的重复情况,遍历过程中若出现a与它前一个值相同时,表明该子问题已经求解过,只需将a往后移动一位即可,同理b,c出现与之前值相同时,也需要(向前向后)移动一位。代码如下:
class Solution {
public:

    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int> > res;
        res.clear();
        int len = nums.size();
        sort(nums.begin(), nums.end());
        for(int i = 0; i < len; ++i){
            if(i > 0 && nums[i] == nums[i-1])
                continue;
            int j = i + 1;
            int k = len -1;
            while(j < k){
                if(j > i + 1 && nums[j] == nums[j-1]){
                    j++;
                    continue;
                }
                if(k < len - 1 && nums[k] == nums[k + 1]){
                    k--;
                    continue;
                }
                int sum = nums[i] + nums[k] + nums[j];
                if(sum < 0)
                    j++;
                else if(sum > 0)
                    k--;
                else{
                    vector<int> tmp;
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[j]);
                    tmp.push_back(nums[k]);
                    res.push_back(tmp);
                    j++;
                }
            }
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值