每週一算:Move Zeros

程式設計師吳師兄發表於2018-10-23

image
leetcode上第283號問題:Move Zeros

給定一個陣列nums,寫一個函式,將陣列中所有的0挪到陣列的末尾,⽽維持其他所有非0元素的相對位置。
舉例: nums = [0, 1, 0, 3, 12],函式運⾏後結果為[1, 3, 12, 0, 0]

解法一

思路:建立一個臨時陣列nonZeroElements,遍歷nums,將nums中非0元素賦值到nonZeroElements中,而後按順序將nonZeroElements賦值到nums上,未遍歷的元素置0;

動畫如下:

image

程式碼如下:

// 時間複雜度: O(n)
// 空間複雜度: O(n)
class Solution {
public:
    void moveZeroes(vector<int>& nums) {

        vector<int> nonZeroElements;

        // 將vec中所有非0元素放入nonZeroElements中
        for(int i = 0 ; i < nums.size() ; i ++)
            if(nums[i])
                nonZeroElements.push_back(nums[i]);

        // 將nonZeroElements中的所有元素依次放入到nums開始的位置
        for(int i = 0 ; i < nonZeroElements.size() ; i ++)
            nums[i] = nonZeroElements[i];

        // 將nums剩餘的位置放置為0
        for(int i = nonZeroElements.size() ; i < nums.size() ; i ++)
            nums[i] = 0;
    }
};

複製程式碼

解法二

思路:設定一個臨時變數k=0,遍歷陣列nums,將非零元素移動到nums[k]位置,同時k++,而後將【k,….nums.size()】中的元素置零。

動畫如下:

image

程式碼如下:

// 原地(in place)解決該問題
// 時間複雜度: O(n)
// 空間複雜度: O(1)
class Solution {
public:
    void moveZeroes(vector<int>& nums) {

        int k = 0; // nums中, [0...k)的元素均為非0元素

        // 遍歷到第i個元素後,保證[0...i]中所有非0元素
        // 都按照順序排列在[0...k)中
        for(int i = 0 ; i < nums.size() ; i ++)
            if(nums[i])
                nums[k++] = nums[i];

        // 將nums剩餘的位置放置為0
        for(int i = k ; i < nums.size() ; i ++)
            nums[i] = 0;
    }
};
複製程式碼

解法三

思路:設定一個臨時變數k=0,遍歷陣列nums,將非零元素與之前的零元素進行交換,維護變數k的值。

動畫如下:

image

程式碼如下:

// 原地(in place)解決該問題
// 時間複雜度: O(n)
// 空間複雜度: O(1)
class Solution {
public:
    void moveZeroes(vector<int>& nums) {

        int k = 0; // nums中, [0...k)的元素均為非0元素

        // 遍歷到第i個元素後,保證[0...i]中所有非0元素
        // 都按照順序排列在[0...k)中
        // 同時, [k...i] 為 0
        for(int i = 0 ; i < nums.size() ; i ++)
            if(nums[i])
                if(k != i)
                    swap(nums[k++] , nums[i]);
                else
                    k ++;
    }
};

複製程式碼

閱讀更多:

在PPT動畫中學演算法