20.4.11 移动零 简单

时间复杂度O(n),空间复杂度O(1)

题目

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

示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]

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

解题思路

  1. 从前往后遍历,遍历时不断更新零的数量,每个数都根据零的数量往前移动相应位置;
  2. 这样操作会把所有0都覆盖掉,在末尾更改相应数量的0。比如零的数量为2,[nums.size()-record, nums.size()]都要改为0。

代码思路

  1. 排除特殊情况;
  2. 变量record记录零的数量;
  3. 循环中,i的边界条件会跟着零的数量进行变化,不然会越界;
  4. 覆盖的时候要考虑用来覆盖的这个数是不是0,通过while的条件判断。如果是0,则要继续往后找不是0的数,并更新零的数量;
  5. while中i+record要小于 nums.size()-1,如果小于nums.size(),下一条语句会越界;
  6. 同时!nums[i+record]很巧妙的解决了第一次和第二次遇到零要不同处理的问题;
  7. 第二个for循环,末尾改零。

代码

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        if(nums.size() == 0 || nums.size() == 1) return;

        int record = 0;
        for(int i = 0; i < nums.size() - record; i++){
            while(i + record < nums.size() - 1 && !nums[i+record]) record++;
            nums[i] = nums[i+record];
        }
        for(int i = nums.size() - record; i < nums.size(); i++)
            if(nums[i] != 0)
                nums[i] = 0;
    }
};
posted @ 2020-04-11 14:50  肥斯大只仔  阅读(98)  评论(0编辑  收藏  举报