题目描述

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

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

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

解题思路

本题解题思路与上一篇的删除有序数组重复元素类似,如下所示:

  • 设定左右指针 i, j,变量 count 用于计数
  • 指针 i 从左向右移动,判断元素是否等于 val;当于 val 相等时 count–, 并移动指针 j,当于 val 不等时指针 i 继续向右移动
  • 当指针 j 移动时,判断元素是否等于 val; 当于 val 相等时 count– 继续移动指针 j, 当不等时互换 i, j 元素并继续移动指针 i
  • 当指针 i,j 交汇时完成退出轮询

可见如下图示:

删除元素

实现

public int removeElement(int[] nums, int val) {
    int i = 0, count = nums.length, j = count;

    while (true) {
        while (i < j) {
            if (nums[i] == val) {
                count--;
                break;
            } else {
                if (i + 1 == j) {
                    break;
                }

                i++;
            }
        }

        while (j > i) {
            j--;

            if (j == i) {
                break;
            }

            if (nums[j] == val) {
                count--;
            } else {
                nums[i] = nums[j];
                i++;
                break;
            }
        }

        if (i >= j) {
            break;
        }

    }

    return count;
}


# LeetCode  

tocToc: