这道题是一道经典的双指针题目,如果暴力解法的话需要两个for循环,但是双指针就能在一个循环里解决。
我先试了试暴力写法,一开始用两个for循环,但是第一个大的循环用for循环少了一些灵活度(因为每次循环都会i+1)有一些问题,改成while后可行
n = len(nums)
k = 0
i=0
while(i<n-k):
if nums[i] == val:
for j in range(i+1, n):
nums[j-1] = nums[j]
k += 1
else:
i+=1
return n - k
首先是考虑循环啥时候结束,因为fast是快的那个指针,它到数组尾部的时候循环就结束了,可以用while fast<n:来判断
我理解的流程是判断fast指向与val不同的值后就硬把值赋值给slow,然后slow和fast都去+1,如果fast指向和val一样的值后就单fast+1
所以就如下所示
while fast<n:
if nums[fast]!=val:
nums[slow]=nums[fast]
slow+=1
fast+=1