python(leetcode)-283移动零

给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。

示例:

输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:

必须在原数组上操作，不能拷贝额外的数组。
尽量减少操作次数。

　说下拿到这道题时的思路：

给人的感觉并不难，首先的想法就是遍历数组中每一个元素，判断如果为0则删除，同时末尾增加0

上代码（通过240ms)击败20%的用户

class Solution:
    def moveZeroes(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        for i in nums:
            if(i==0):
                nums.remove(i)
                nums.append(0)
        

if __name__=="__main__":
    s=Solution()
    nums=[0,1,0,3,12]
    print(s.moveZeroes(nums))

代码非常简洁，只有短短4行，但是对比其他方法效率却不高，

分析代码的时间复杂度

外层for循环需要N次，remove(i)需要N次，append()方法1次

所以时间复杂度为O(n^2)

换一种方法，上代码(通过) 56ms 击败99%

class Solution:
    def moveZeroes(self,nums):
        """

        :param nums:
        :return:
        """
        count=0
        zero=0
        for i in range(len(nums)):
            if(nums[i]!=0):           #判断是否为0
                nums[count]=nums[i]   #不是0的数向前移
                count+=1              #移动一个 计数加一
            else:
                zero+=1
        for j in range(count,len(nums)): #把最后位置补0
            nums[j]=0
        return nums
if __name__=="__main__":
    s=Solution()
    nums=[0,1,0,3,12]
    print(s.moveZeroes(nums))

 思路在代码中有注释

分析下时间复杂度：

for循环有N次，if语句1次，赋值语句1次，++1次

第二个for循环N次，赋值语句1次

两个for循环是并列关系 所以时间复杂度为O(n) 可以发现确实速度快了很多