C++每日一题

题目:给定一个数组 nums 和一个值 val ,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成,OJ链接.

法一:暴力法

解析:
我们来解析一下这个题目的做题思路,他的含义就是让我们删除掉数组中的元素,然后将数组后面的元素跟上来。最后返回删除掉元素的数组长度即可。比如数组长度为 10,里面有2个目标值,我们最后返回的长度为 8,但是返回的 8 个元素,需要排在数组的最前面。那么暴力解法的话则就需要两个 for 循环,一个用来找到删除,另一个用来更新数组。
C++每日一题_第1张图片
总体思路就是这样的,后面的会不断往前覆盖。暴力解法也是不超时的,实现也不算太简单主要需要注意两个地方。

(1)需要先定义变量len获取数组长度,因为后面我们的返回的数组长度是改变的,所以不可以用 nums.length 作为上界

(2)我们每找到一个需要删除的值的时候,需要i–,防止出现多个需要删除的值在一起的情况,然后漏删。
题目代码

class Solution {
     
public:
    int removeElement(vector<int>& nums, int val) {
     
        int len = nums.size();
        for(int i = 0; i<len; i++)
        {
     
            if(nums[i] == val)
            {
     
                for(int j = i;j<len-1;j++)
                {
     
                    nums[j] = nums[j+1];
                }
                i--;
                len--;
            }
        }
        return len;
    }
};

法二:双指针

快慢指针的做法只需要一个 for 循环即可解决,时间复杂度为 O(n) ,总体思路就是有两个指针,前面一个后面一个,前面的用于搜索需要删除的值,当遇到需要删除的值时,前指针直接跳过,后面的指针不动,当遇到正常值时,两个指针都进行移动,并修改慢指针的值。最后只需输出慢指针的索引即可。

// 时间复杂度: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;
    }
};

你可能感兴趣的:(C++,指针,leetcode,数据结构,算法)