LeetCode-数组-删除元素

题目描述

给定一个数组 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;
}