五分鐘搞定一道算法題:Move Zeros

image

leetcode上第283號問題:Move Zeros

給定一個數(shù)組nums,寫一個函數(shù),將數(shù)組中所有的0挪到數(shù)組的末尾,?維持其他所有非0元素的相對位置。
舉例: nums = [0, 1, 0, 3, 12],函數(shù)運?后結(jié)果為[1, 3, 12, 0, 0]

解法一

思路:創(chuàng)建一個臨時數(shù)組nonZeroElements,遍歷nums,將nums中非0元素賦值到nonZeroElements中,而后按順序?qū)onZeroElements賦值到nums上,未遍歷的元素置0;

動畫如下:

image

代碼如下:

// 時間復(fù)雜度: O(n)
// 空間復(fù)雜度: 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;
    }
};

解法二

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

動畫如下:

image

代碼如下:

// 原地(in place)解決該問題
// 時間復(fù)雜度: O(n)
// 空間復(fù)雜度: 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;
    }
};

解法三

思路:設(shè)定一個臨時變量k=0,遍歷數(shù)組nums,將非零元素與之前的零元素進行交換,維護變量k的值。

動畫如下:

image

代碼如下:

// 原地(in place)解決該問題
// 時間復(fù)雜度: O(n)
// 空間復(fù)雜度: 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 ++;
    }
};

敬請關(guān)注

image
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容