LeetCode-数组-删除元素

34次阅读

共计 607 个字符,预计需要花费 2 分钟才能阅读完成。

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

正文完
 0