1. 程式人生 > >Leetcode 移動零(c++)

Leetcode 移動零(c++)

給定一個數組 nums,編寫一個函式將所有 0 移動到陣列的末尾,同時保持非零元素的相對順序。

示例:

輸入: [0,1,0,3,12]
輸出: [1,3,12,0,0]
說明:

必須在原陣列上操作,不能拷貝額外的陣列。
儘量減少操作次數。

解:用遞減iterator刪除0(erase),記錄刪除的0個數,最後在陣列後面插入n個0.

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int cnt=0;
        vector<int>::iterator it=
nums.end()-1; for (;it>=nums.begin();--it){ if(*it == 0){ nums.erase(it); cnt += 1; } } nums.insert(nums.end(),cnt,0); } };