【Leetcode刷题笔记】27. 移除元素

原题链接

Leetcode 27. 移除元素

题目

给你一个数组 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,3,0,4]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

题目分析

思路一：直接遍历数组，把等于 val 的数删除，并把后面的数往前挪动一位，最后输出新数组的长度

缺点：时间复杂度过高，而且频繁挪动大量数字的行为并不明智

思路二：使用快慢指针法。创建两个变量 src 和 dest，我们先让 src++，当 src 上的数等于 val 时，就继续 ++；当 src 上的数不等于 dest 时，我们就把 src 的数赋值给 dest 上的数，两个都 ++，直到 src 遍历完数组。此时 dest 的值就是新数组的长度，数组内等于 val 的数也已经被覆盖掉了。这样子我们就只遍历了一遍数组，时间复杂度很小，属于优解。

示意图：

代码实现

// 快慢指针法
#include <stdio.h>

int removeElement(int* nums, int numsSize, int val) {
    // 先定义两个变量
    int src = 0;
    int dest = 0;
    // src先走，而且src的长度要被数组长度限制
    while (src < numsSize) {
        // src位置所在的数等于val，就继续往前走
        if (nums[src] == val) {
            src++;
        }
        // src不等于val，就把src上的值赋给dest，两个再一起往前走
        else {
            nums[dest] = nums[src];
            dest++;
            src++;
        }
    }
    return dest;
}

int main()
{
    int arr[] = { 3,2,2,3 };
    int numsSize = sizeof(arr) / sizeof(arr[0]);
    int len = removeElement(arr, numsSize, 3);
    int i = 0;
    printf("新数组：");
    for (i = 0; i < len; i++)
    {
        printf("%d ", arr[i]);
    }
    printf("\n新数组的长度：%d\n", len);
    return 0;
}

运行结果如下

 ​​​​​​​