【c++】LeetCode283. 移动零

LeetCode283. 移动零

给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]

---

方法一：
创建一个临时数组 noZeroVec，将原数组 nums 中的非零元素复制过去，再把noZeroVec中的元素覆盖掉nums前面一段，最后把nums后面的元素置零便可。
时间复杂度：O（n）
空间复杂度：O（n）

void moveZeroes(vector<int>& nums) {
vector<int> noZeroVec;

//将原数组 nums 中的非零元素复制过去
for(int i = 0; i < nums.size(); i++)
    if(nums[i])
        noZeroVec.push_back( nums[i] );

//将所有非零元素移回nums
for(int i = 0; i < noZeroVec.size(); i++)
    nums[i] = noZeroVec[i];

//把nums剩下未被覆盖的元素置零
for(int i = noZeroVec.size(); i < nums.size(); i++)
    nums[i] = 0;
}

方法二：
其实我们完全可以不使用额外的空间。
设一个指针k，遍历数组，在[0-k)中保存遍历过的非零元素。最后把k后面的元素置零。
时间复杂度：O（n）
空间复杂度：O（1）

void moveZeroes(vector<int>& nums) {
    int k = 0; //[0-k)中的元素为非零元素，[0)区间中不存在元素

    // 遍历数组，把非零元素放在[0-k)中
    for(int i = 0; i < nums.size(); i++)
        if( nums[i] )
            nums[k++] = nums[i];

    // 把nums剩余的元素置零
    for(int i = k; i < nums.size(); i++)
        nums[i] = 0;
}

方法三（法二的优化版）：
根据方法二的思想，还要把后面的元素依次赋为0，感觉不太爽。既然数组中有0，为何不在元素放在[0-k）的同时让0移到后面？

k的定义不变。遍历数组。
i 遇到 0 时，只需指针 i 向后移一。
i 遇到 非零元素 时，交换元素nums[k] 和 nums[i] ；指针 i，k 向后移一。此时区间 [ k - i ] 全为0。

优化：在未遇到第一个0前，i = k。此时只需让 i，k向后移一便可。

时间复杂度：O（n）
空间复杂度：O（1）

void moveZeroes(vector<int>& nums) {
int k = 0; //[0-k)中的元素为非零元素，从k是第一个0的位置

// 遍历数组，交换零和非零元素的位置
// 同时[k-i]为0
for(int i = 0; i < nums.size(); i++)
    if( nums[i]) 
        if( i != k ) 
            swap( nums[k++], nums[i] );
        else // 当数组中没有0的时候，就不用每一次都交换了
            k++;
}