公众号:爱写bug
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
Given an array nums and a value val, remove all instances of that value in-place and return the new length.
Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.
The order of elements can be changed. It doesn't matter what you leave beyond the new length.
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
Confused why the returned value is an integer but your answer is an array?
Note that the input array is passed in by reference, which means modification to the input array will be known to the caller as well.
Internally you can think of this:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
解题思路:
只允许原数组修改,可以用双指针,左指针 i 自左向右,右指针 j 从右向左。如果索引 i 和 val 相等,则索引 i 得到索引 j 的值,并且 j 前移一位。如果不相等,i 后移一位。
因为要求是返回修改后的长度并只考虑该长度的数组,那么就不用考虑该长度之后的数组,所以只需得到索引 j 的值,不用再把索引 j 的值改为索引 i的值。
Java:
class Solution {
public int removeElement(int[] nums, int val) {
int i=0,j=nums.length-1;//i-左指针;j-右指针
while (i<=j){
if(nums[i]==val){
nums[i]=nums[j];//得到索引j的值,无需把索引j的值改为索引i的值
j--;
}else i++;
}
return j+1;
}
}
Python3:
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
i=0
j=len(nums)-1
while i<=j:
if(nums[i]==val):
nums[i]=nums[j]
j-=1
else:i+=1
return j+1
总结:
这道题本身很简单,只要搞清思路,一起都会变得明了。