STL中vector的size()返回类型为size_t造成的for循环判断条件下溢问题

在刷LeetCode-15三数之和发现了一个vector.size()的返回值size_t类型可能引发的问题,之前看到过有人规避过这个问题单一直没有亲身碰到过也就不以为然,今天终于见识了,LeetCode调试器调试了半天才发现问题所在- - !。
首先是有问题的代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.empty()) return res;
        // 排序一下在做
        sort(nums.begin(), nums.end());
        if(nums[0]>0 || nums[nums.size()-1]<0) return res;

        // i<nums.size()-2是因为最后要留两个数来相加
        for(int i=0; i<(nums.size()-2; ++i){    // ### 1
            if(nums[i]>0) break; // 第一个元素为正数之后就不需要再遍历了
            if(i>0 && nums[i]==nums[i-1]) continue; // 避免找到重复的
            int left=i+1, right=nums.size()-1;
            while(left<right&&nums[right]>=0){
                if(nums[left]+nums[right] == -nums[i]){
                    res.push_back({nums[i], nums[left], nums[right]});
                    // 更新位置,跳过重复的数
                    while(left<right && nums[left]==nums[left+1]) left++;
                    while(left<right && nums[right]==nums[right-1]) right--;
                    left++, right--;
                }
                else if(nums[left]+nums[right] < -nums[i]){
                    left++;
                } 
                else{
                    right--;
                }
            }
        }
        return res;
    }
};

问题发生在for循环的条件处(###1),判断条件为i<nums.size()-2,当nums.size()-x>=0时不会出现问题,但当nums.size()-x<0问题就发生了,由于nums.size()的返回值类型为size_t是一个无符号类型整数,因而当nums.size()-x<0时会发生无符号整数的下溢,从而变成一个很大的正整数,也就使得循环成为了一个“死循环”。
解决方案:在nums.size()前加上一个强制类型转换,将size_t类型转换为int类型,这样就会得到负数,从而不会进入循环,正确代码如下:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.empty()) return res;
        // 排序一下在做
        sort(nums.begin(), nums.end());
        if(nums[0]>0 || nums[nums.size()-1]<0) return res;

        // i<nums.size()-2是因为最后要留两个数来相加
        // nums.size()是size_t无符号整型,当-2为负数后溢出后变为正整数上限
        for(int i=0; i<(int)nums.size()-2; ++i){
            if(nums[i]>0) break; // 第一个元素为正数之后就不需要再遍历了
            if(i>0 && nums[i]==nums[i-1]) continue; // 避免找到重复的
            int left=i+1, right=nums.size()-1;
            while(left<right&&nums[right]>=0){
                if(nums[left]+nums[right] == -nums[i]){
                    res.push_back({nums[i], nums[left], nums[right]});
                    // 更新位置,跳过重复的数
                    while(left<right && nums[left]==nums[left+1]) left++;
                    while(left<right && nums[right]==nums[right-1]) right--;
                    left++, right--;
                }
                else if(nums[left]+nums[right] < -nums[i]){
                    left++;
                } 
                else{
                    right--;
                }
            }
        }
        return res;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值