LeetCode(15)--3Sum

来源:互联网 发布:苹果手机模拟器mac版 编辑:程序博客网 时间:2024/05/24 15:42

题目:找到一组数字中,三个数之和为0的所有数。值得注意的就是不能有重复的组合。
思路:
如果是暴利搜索肯定会超时,所以首先对原数组进行排序,然后开始遍历排序后的数组,这里注意不是遍历到最后一个停止,而是到倒数第三个就可以了,中间如果遇到跟前一个数相同的数字就直接跳过。对于遍历到的数,如果大于0则跳到下一个数,因为三个大于0的数相加不可能等于0;否则用0减去这个数得到一个sum,我们只需要在这个数之后找到两个数之和等于sum即可,这样一来问题又转化为了求two sum,这时候我们一次扫描,找到了等于sum的两数后,加上当前遍历到的数字,按顺序存入结果中即可,然后还要注意跳过重复数字。时间复杂度为 O(n2)。实现代码如下:

class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        vector<vector<int>> res = {};        if (nums.size() < 3) return res;        sort(nums.begin(), nums.end());        for (int i = 0; i < nums.size() - 2; i++)        {            if (nums[i]>0)//如果当前位大于0,排序后后面也大于0,三个数之和不会等于0                break;            if (i > 0 && nums[i] == nums[i - 1]){ //当前位和前一位相同时,省去继续寻找                continue;            }            int sum = -nums[i];            int left = i + 1, right = nums.size() - 1;            while (left < right)            {                if (nums[left] + nums[right] < sum)                {                       while ((left < right) && (nums[left++] == nums[left]))//跳过重复的数字                    {                    }                }                else if (nums[left] + nums[right] > sum)                {                    while ((left < right) && (nums[right--] == nums[right]))//跳过重复的数字                    {}                }                else                {                    vector<int> three;                    three.push_back(nums[i]);                    three.push_back(nums[left]);                    three.push_back(nums[right]);                    res.push_back(three);                    while ((nums[left ++] == nums[left])&&(nums[right--]==nums[right]))                    {                    }                }            }        }        return res;    }};
原创粉丝点击