80.每日一练:移除元素(力扣)

问题描述

80.每日一练:移除元素(力扣)_第1张图片

80.每日一练:移除元素(力扣)_第2张图片

代码解决以及思想 

解法一

class Solution {
public:
    int removeElement(vector& nums, int val) {
        int len = 0; // 初始化一个用于记录非目标值个数的变量

        // 创建一个迭代器 it,指向 nums 的开头
        vector::iterator it = nums.begin();

        // 遍历 nums
        while (it != nums.end()) {
            if (*it == val) {
                // 如果当前元素等于目标值 val,需要将其移除
                nums.erase(it); // 使用 erase 函数移除元素
            } else {
                // 如果当前元素不等于目标值 val,增加非目标值的计数 len,并将迭代器 it 向前移动
                len++;
                it++;
            }
        }

        return len; // 返回新长度
    }
};

代码的基本思想是使用一个迭代器 it 遍历 nums 中的元素,同时使用变量 len 来记录非目标值的元素的个数。具体步骤如下:

  1. 初始化 len 为0,用于记录非目标值的元素个数。

  2. 使用迭代器 it 初始化为 nums 的开头(begin())。

  3. 进行循环,检查 it 是否等于 nums 的结束位置(end())。

  4. 在循环内部,检查 it 指向的元素是否等于给定的目标值 val

    • 如果相等,说明需要移除该元素,使用 nums.erase(it) 来移除当前元素,这会使vector的大小减小。注意,这里没有将 it 自增,因为在移除元素后,it 自动指向下一个元素。

    • 如果不相等,说明当前元素是非目标值,将 len 自增并将 it 自增,移向下一个元素。

  5. 循环结束后,len 将表示非目标值的元素个数。vector nums 中已经被修改,移除了所有值等于 val 的元素。

  6. 返回 len,表示修改后的vector的新长度。

解法二 

class Solution {
public:
    int removeElement(vector& nums, int val) {
        int left = 0;         // 左指针,指向数组开头
        int right = nums.size();  // 右指针,指向数组末尾

        // 循环直到左指针小于右指针
        while (left < right) {
            if (nums[left] == val) {
                // 如果左指针指向的元素等于目标值 val,将左指针元素替换为右指针前面的元素
                nums[left] = nums[right - 1];
                right--;  // 右指针向左移动,排除一个目标值
            } else {
                // 如果左指针指向的元素不等于目标值 val,增加左指针,继续遍历
                left++;
            }
        }

        return left;  // 返回新长度
    }
};

你可能感兴趣的:(刷题,1024程序员节)