LeetCode 筆試題 Move Zeroes（將0移動到末尾）

問題描述 

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

示例:

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

說明:

1. 必須在原數組上操作，不能拷貝額外的數組。
2. 儘量減少操作次數。

代碼1.

思路：把所有的0移到最後面 = 把所有不是0的數與此時數組最前面的0交換位置。

具體做法：兩個指針，遍歷整個數組：

第一個指針之前的數都不是0（一般情況它本身指的數是0）；

第一個指針和第二個指針中間的數都是0；

這個時候只要第二個指針指向的數不是0，就可以交換兩個指針指向的值。

 

void moveZeroes(vector<int>& nums)
{
	for (int beforeItNotZero = 0, cur = 0; cur < nums.size(); cur++)
	{
		if (nums[cur] != 0) 
		{
			swap(nums[beforeItNotZero], nums[cur]);
			beforeItNotZero++;
		}
	}
}

 

代碼2.

思路：把0刪了，記錄0的個數，再加在最後面。

//!!!!! 這裏使用了迭代器進行erase，需要對迭代器進行更新防止迭代器失效
void moveZero(vector<int>&nums)
{
	auto it = nums.begin();
	int count = 0;
	while (it != nums.end())
	{
		if (*it != 0)
		{
			it++;
		}
		else
		{
			it = nums.erase(it);//注意迭代器需要更新
			count++;
		}
	}
	for (int i = 0; i < count; i++)
	{
		nums.push_back(0);
	}

}