LeetCode面试150——27移除元素_更改 nums 数组,使nums的前k个元素包含不等于val的元素-CSDN博客

本文链接：https://blog.csdn.net/Junmo12345/article/details/140665001

题目难度：简单

默认优化目标：最小化平均时间复杂度。

Python默认为Python3。

1 题目描述

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。

用户评测：

评测机将使用以下代码测试您的解决方案：

int[] nums = [...]; // 输入数组
int val = ...; // 要移除的值
int[] expectedNums = [...]; // 长度正确的预期答案。
                            // 它以不等于 val 的值排序。

int k = removeElement(nums, val); // 调用你的实现

assert k == expectedNums.length;
sort(nums, 0, k); // 排序 nums 的前 k 个元素
for (int i = 0; i < actualLength; i++) {
    assert nums[i] == expectedNums[i];
}

如果所有的断言都通过，你的解决方案将会通过。

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2,_,_]
解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3,_,_,_]
解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

提示：

0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100

2 题目解析

这道题的输入是一个数组nums和一个值val，输出是nums中不等于val的元素数量k以及在前k个不等于val的元素的更新后的数组nums。

题目描述中有这么一句话，"nums的其余元素和nums的大小并不重要"。并且结合"你需要原地移除所有数值等于val的元素"，我们要做的操作是将nums中元素值等于val的元素用不等于val的值给他替换掉，并统计这样的元素的个数k。不用执行排序、删除等操作。

3 算法原理及程序实现

3.1 双指针

元素的移动很自然能想到双指针。和88合并两个有序数组的双指针法不同，这里是有两个指针指向nums，两个指针的初始位置都指向nums的头节点。一个用于遍历nums，一个用于定位不等于val的元素的位置，即跳过值等于val的元素。跳过的个数就等于k。

平均时间复杂度为O(n)，即遍历一遍数组。平均空间复杂度为O(1)，不需要额外开辟新的数组。

C++代码实现

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int left=0;//指针，用于跳过值为val的元素
        int n=nums.size();//nums的长度，用一个变量记录，这样在循环条件里不用仿佛计算浪费时间了

        for(int right=0;right<n;right++){//right用于遍历数组
            if(nums[right]!=val){
                nums[left]=nums[right];
                left++;
            }
        }

        return left;

    }
};

Python代码实现

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        left=0
        n=len(nums)

        for right in range(n):
            if nums[right]!=val:
                nums[left]=nums[right]
                left+=1
        return left

3.2 优化双指针

在3.1的基础上，我们将指针改为：一个指向头节点，一个指向尾节点，然后像中间遍历。如果遇到等于val的元素，跳过它。这样相当于只要遍历原来数组长度的一半。

平均时间复杂度为O(n)，平均空间复杂度为O(1)。

在3.1的代码上进行修改

C++代码实现

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int left=0;//指针，指头
        int right=nums.size()-1;//指针，指尾

        while(left<=right){//两指针遇到停止
            if(nums[left]==val){
                nums[left]=nums[right];
                right--;//指针左移
            }
            else{
                left++;//指针右移
            }
        }

        return left;

    }
};

Python代码实现

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        left,right=0,len(nums)-1

        while left<=right:
            if nums[left]==val:
                nums[left]=nums[right]
                right-=1
            else:
                left+=1

        return left