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