我的第一反应是删除掉所有val值
我用到了双指针,根据while循环+if判断语句
遍历数组中判断是否为val,如果是就删除
这里不得不提到while和for的区别
由于删除的过程中数组的索引号就发生了改变
如果是for,那么i是每次循环都加1,会导致混乱
但用while的话,可以动态的决定要不要让i加1
这就能避免因删除导致的混乱
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
n=len(nums)
i,j=0,0
while i<n-j:
if nums[i]==val:
nums.remove(nums[i])
j=j+1
else:
i=i+1
return n-j
后来再看看题,好像没有要求删除,只是要求前几项。
我就不把判断重心放在是val上了
而放在不是val上了
把所有不是val的值拿出来放在前面
不删除改成覆盖
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
n=len(nums)
j=0
for i in range(n):
if nums[i]!=val:
nums[j]=nums[i]
j=j+1
return j
我用C++写了一遍同样的覆盖版代码
只能说C++真香
无论是执行用时还是内存消耗都远远小于python
并且执行用时击败比例也远远大于python的
class Solution {
public:
int removeElement(vector<int>& nums, int val)
{
int n=nums.size();
int j=0;
for(int i=0;i<n;i++)
if(nums[i]!=val)
nums[j++]=nums[i];
return j;
}
};