题目描述

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
不要使用额外的数组空间，你必须仅使用 O(1)额外空间并原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

样例

给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。 ！！！

算法1

只需要每次判断 nums[i] 与 val 是否相同；若相同，则 nums[k] 赋值为 nums[i] 并且 k++。

时间复杂度

由于仅有一次遍历，故时间复杂度为 O(n)。

C 代码

int removeElement(int* nums, int numsSize, int val){
    if (nums == NULL || numsSize == 0) {
        return 0;
    }
    int k = 0;
    for (int i = 0; i < numsSize; i ++ ) {
        if (nums[i] != val) {
            nums[k ++ ] = nums[i];
        }
    }
    return k;
}

算法2

时间复杂度

O(n2)

空间复杂度

O(1)

C 代码

/*
读题没有细致 
原地排序：空间复杂度为O(1)的排序算法
稳定排序：排序后相对位置不发生改变
*/
int removeElement(int* nums, int numsSize, int val){
    if (nums == NULL || numsSize == 0) {
        return 0;
    }

    while (numsSize > 0) // 边界问题多了 10分钟
    {
        if (nums[numsSize - 1] == val) {
            numsSize --;
        }
        else {
            break;
        }
    }
    if (numsSize == 0) {
        return 0;
    }
     printf("%d", numsSize);
    // return numsSize;
    int t = 0, len = numsSize;
    for (int i = numsSize - 1; i >= 0; i -- ) {
        if (nums[i] == val) {
            len -- ;
            for (int j = i; j < numsSize - 1; j ++ ) { // 需要-1要不数组越界 10分钟
                t = nums[j];
                nums[j] = nums[j + 1];
                nums[j + 1] = t;
            }
        }
    }
    return len;
}