LeetCode (15)3Sum

31 篇文章 0 订阅

(15)3Sum

题目:在一个数组里面找到三个数字和等于0,输出所有的可能性,不要存在重复的组。

例子:提供的数组:s=[-1, 0, 1, 2, -1, -4],返回的结果应该是:[ [-1, 0, 1], [-1, -1, 2]]。

这道题本来想按原来两个数相加的那个来算,引用map,但是后来发现无法获取是否前面已经输出同样的结果,所以就很麻烦,或许有解决方法,但是暂时觉得可能会兜圈子,不如直接使用简单一点的排序方法。

首先直接sort一下vector,每次抽走最前面的一个数字,然后查找剩余的数组内是否有可以和为target=(0-抽走数字)的组合,通过left和right的查找方法,因为已经排好序了,如果和小于target,那么就应该left右移,这样能使得和变大,否则right左移,使得和变小,找到之后就记录一条,直到所有都记录完成。

避免重复的方法就显而易见了,由于都已经排序好了,那么每次找的时候只需要在一段子数组内有重复的跳过就好了,比如上面left右移和right左移的时候,只要是下一个跟当前数字相同,那么不运算直接移动一次,直到下一个数字和当前数字不同,才将当前数字放入运算中。

另外,如果暴力的三层循环搜索也就是O(n^3)的方法会导致超时TLE。

下面是代码:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int len = nums.size();
        vector<vector<int>> s;
        if(len == 0){
            return s;
        }
        sort(nums.begin(), nums.end());
        int target = 0;
        int i, j, k;
        vector<int> temp;
        for(i=0;i<len;i++){
            if(i!=0){
                if(nums[i] == nums[i-1]){
                    continue;
                }
            }
            target = 0;
            target -= nums[i];
            j = i + 1;
            k = len - 1;
            while(j<k){
                if(nums[j] + nums[k] == target){
                    temp.push_back(nums[i]);
                    temp.push_back(nums[j]);
                    temp.push_back(nums[k]);
                    s.push_back(temp);
                    temp.clear();
                    while(j<k&&nums[j]==nums[j+1]){
                        j ++;
                    }
                    while(j<k&&nums[k]==nums[k-1]){
                        k --;
                    }
                    j++;
                    k--;
                }
                else if(nums[j] + nums[k] < target){
                    j ++;
                }
                else{
                    k --;
                }
            }
        }
        return s;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值