给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums =[0,1,0,3,12]
输出:[1,3,12,0,0]
示例 2:
输入: nums =[0]
输出:[0]
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
思路:不考虑太多,很直观的解法就是两层for循环,如果元素是0,则把它往后移动,但这样的解法,时间复杂度为O(n^2),不是一种好的解决办法。
这道题其实和27.移除元素很相像,要求在不复制数组的情况下原地对数组进行操作,把所有的0元素移到数组末尾,那我们可以仅遍历一边数组,k是代表非0元素的个数,同时也是下标,若该元素不等于0,则赋值给nums[k],遍历完一边数组之后,nums的前k个元素一定都不等于0,且满足题目保持非零元素的相对顺序这个要求,然后我们将数组从k到len(nums)-1全部赋值为0,就巧妙地完成了把所有0移到末尾的操作。
27.移除元素Leetcode链接:. - 力扣(LeetCode)
class Solution(object):
def moveZeroes(self, nums):
k = 0
for i,num in enumerate(nums):
if num != 0:
nums[k] = num
k += 1
for i in range(k,len(nums)):
nums[i] = 0
return nums