Bug01

之夏797

已于 2023-09-21 19:44:23 修改

阅读量41

点赞数

分类专栏： # BUG 文章标签： c++ 算法

于 2023-09-21 19:43:50 首次发布

本文链接：https://blog.csdn.net/qq_63958683/article/details/133145680

版权

BUG 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

文章讲述了作者在解决LeetCode题目时遇到的内存溢出问题，通过分析错误日志和代码，发现是由于数组越界导致的heap-buffer-overflow。随后，作者提出并优化了使用双指针的方法，解决了问题，提高了代码效率。

摘要由CSDN通过智能技术生成

Bug01

ERROR: AddressSanitizer: heap-buffer-overflow on address 0x6020000000a0 at pc 0x0000003677a5 bp 0x7ffd110a30b0 sp 0x7ffd110a30a8

Leetcode题目

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

我的代码是：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int p = 0, prep = 0;
        int count = 0;
        while(prep < nums.size())
        {
            if(nums[prep] == val)
            {
                prep++;
                count++;
            }
            nums[p] = nums[prep];
            prep++;
            p++;
        }
        return nums.size() - count;
    }
};

编译无法通过，但并没有指出哪一行的问题。

经过网上查阅资料，得知这是常见的数组越界问题。数组的索引应该不超过数组的大小。

检查之后，发现我的方法在循环里的做法存在问题。

先通过判断语句，判断快指针是否指到了要删除的val，如果是，就会将快指针后移一个位置，再将此时快指针的位置的值赋值给慢指针位置的数组元素。

则存在一个问题：当要删除的val正好在数组的最后一个位置，快指针扫描到它之后，后移一个位置，再进行赋值动作时就将超过数组的最大索引，发生越界。

修改后的代码为：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int p = 0, prep = 0;
        int count = 0;
        while(prep < nums.size())
        {
            if(nums[prep] == val)
            {
                prep++;
                count++;
            }
            if(prep < nums.size() && nums[prep] != val)
            {
                nums[p] = nums[prep];
                prep++;
                p++;
            }
        }
        return nums.size() - count;
    }
};

可以参考代码随想录中的双指针法移除元素,更加精妙。

// 时间复杂度：O(n)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
            if (val != nums[fastIndex]) {
                nums[slowIndex++] = nums[fastIndex];
            }
        }
        return slowIndex;
    }
};