15. 3Sum(M)

题目描述

给定一个长度为n的数组,找出3个数a,b,c,使之满足a+b+c=0这个等式。找出满足这个条件的所有不重复集合。原题以及例子如下,
这里写图片描述

题目分析

PS:答案的集合不要求是有序的。

如果是暴力破解的话,那么这道题的时间复杂度为O(n^3),穷举出所有的例子,但肯定是会超时的。我们可以做出这么一个转换,a+b=-c。做过Two Sum的朋友,肯定一下子就能想到如何以O(n)的时间复杂度来解决这个问题。那么这道题便迎刃而解了。
我在这里简单介绍下,如何以O(n)的时间复杂度解决a+b=-c这个问题。有两种方法,一是直接使用STL中的HashMap,这个我就不介绍了,相信大家看了讨论区的答案一下就知道了;二是使用两个指针的方法。在这里我们假定已知的数组已经是有序的(这里的时间复杂度为O(n^2),所以可以在开始之间进行一个快排)。用两个指针,分别指向数组的头和尾,如果tail+head<-c,那么便把头指针想后移动;如果tail+head>-c,那么就把尾指针向前移动。显然,当我遍历完该数组后,便可得到所有的答案。此题解答源码如下,

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());

        vector<vector<int> > res;

        bool findOne = false;
        int l = nums.size();
        for (int i = l - 1; i >= 2; --i) {
            if (i != l - 1 && nums[i] == nums[i + 1]) {
                continue;
            }
            int goal = -nums[i];
            findOne = false;
            int first = 0;
            int last = i - 1;
            int preFirst = 0;
            while (first != last) {
                if (nums[first] + nums[last] < goal) {
                    ++first;
                } else if (nums[first] + nums[last] > goal) {
                    --last;
                } else {
                    if (findOne && (preFirst == nums[first])) {
                        ++first;
                        continue;
                    }
                    vector<int> temp = {nums[first], nums[last], nums[i]};
                    res.push_back(temp);
                    preFirst = nums[first];
                    ++first;
                    findOne = true;

                }
            }
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值