Leetcode 15 3Sum + 有趣的小BUG

triplebee

发布于 2018-01-12 15:10:18

6330

发布于 2018-01-12 15:10:18

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: 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]
]

题目不难，意外发现一个容易被忽视的C++小Tips。

题意为找出和为0的三元组，按升序排列，不重复。

首先肯定是排序，

然后我习惯性想到枚举两个数，最后一个用map或者hash来表示是否存在，

然而这样不是和之前某两个数求和的问题一样了吗？

Leetcode显然不会这么无聊，于是我想了很久，无意中发现两点法可行，

复杂度和第一种用hash的情况一样，都为n方，但是实现起来要巧妙很多，

枚举第一个数，以第一个数后边的一个数和末尾的数为两点，往中间缩，

如果三者的和大于零，说明最大值需要小一些，因为最小值已经无法更小了，就将末尾前移，

反之，则将头部后移。

等于零的时候记录一下，继续找。

写好后调试了一会儿，发现了无法解释的BUG，

nums.size()-2     这一句单独拿出来，值非常奇怪，需要借助int变量才能变回去！

我搞ACM的时候一直是这么写的啊！

后来经过观察，我发现vector size()的返回值为一个无符号整形，

因为ACM题的数据范围一般不会让它出现负数的情况，所以我一直都没有发现，

而这里-2显然在输入为空的时候，会让无符号整形溢出，好的，问题解决

又学到了一个小细节。

详情请看代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        int len=(nums.size()-2);
        for(int i=0;i<len;i++)
        {
            //cout<<i<<" "<<nums.size()-2<<endl;  就是这里RE发现问题
            if(nums[i]>0)continue;
            if(i!=0 && nums[i]==nums[i-1])continue;
            int l=i+1,r=nums.size()-1;
            while(l<r)
            {
                int com=nums[i]+nums[l]+nums[r];
                if(com>0)
                    r--;
                else if(com<0)
                    l++;
                else
                {
                    result.push_back({nums[i], nums[l], nums[r]});
                    while(l<r && nums[l]==nums[l+1]) l++;
                    while(l<r && nums[r]==nums[r-1]) r--;
                    r--;
                    l++;
                }
            }
        }
        return result;
    }
};

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2016-08-28 ，如有侵权请联系 cloudcommunity@tencent.com 删除

hash