給定一個陣列nums,寫一個函式,將陣列中所有的0挪到陣列的末尾,⽽維持其他所有非0元素的相對位置。
舉例: nums = [0, 1, 0, 3, 12],函式運⾏後結果為[1, 3, 12, 0, 0]
解法一
思路:建立一個臨時陣列nonZeroElements,遍歷nums,將nums中非0元素賦值到nonZeroElements中,而後按順序將nonZeroElements賦值到nums上,未遍歷的元素置0;
動畫如下:
程式碼如下:
// 時間複雜度: 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()】中的元素置零。
動畫如下:
程式碼如下:
// 原地(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的值。
動畫如下:
程式碼如下:
// 原地(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 ++;
}
};
複製程式碼
閱讀更多: