LeetCode 27. 移除元素

LeetCode 27. 移除元素

文章目录

一、题目

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

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

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

说明:

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

二、C# 题解

很简单的一道题,直接附上代码:

public class Solution {
    public int RemoveElement(int[] nums, int val) {
        int i = 0, j = 0;
        while (true) {
            while (j < nums.Length && nums[j] == val) j++; // 找到下一个不为 val 的元素
            if (j == nums.Length) break;                   // j 越界则结束循环
            nums[i++] = nums[j++];                         // 赋值并后移 i、j
        }
        return i;
    }
}
  • 时间复杂度:O ( n ) O(n)O(n)。
  • 空间复杂度:O ( 1 ) O(1)O(1)。