LeetCode283. 移动零
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
方法一:
创建一个临时数组 noZeroVec,将原数组 nums 中的非零元素复制过去,再把noZeroVec中的元素覆盖掉nums前面一段,最后把nums后面的元素置零便可。
时间复杂度:O(n)
空间复杂度:O(n)
void moveZeroes(vector<int>& nums) {
vector<int> noZeroVec;
//将原数组 nums 中的非零元素复制过去
for(int i = 0; i < nums.size(); i++)
if(nums[i])
noZeroVec.push_back( nums[i] );
//将所有非零元素移回nums
for(int i = 0; i < noZeroVec.size(); i++)
nums[i] = noZeroVec[i];
//把nums剩下未被覆盖的元素置零
for(int i = noZeroVec.size(); i < nums.size(); i++)
nums[i] = 0;
}
方法二:
其实我们完全可以不使用额外的空间。
设一个指针k,遍历数组,在[0-k)中保存遍历过的非零元素。最后把k后面的元素置零。
时间复杂度:O(n)
空间复杂度:O(1)
void moveZeroes(vector<int>& nums) {
int k = 0; //[0-k)中的元素为非零元素,[0)区间中不存在元素
// 遍历数组,把非零元素放在[0-k)中
for(int i = 0; i < nums.size(); i++)
if( nums[i] )
nums[k++] = nums[i];
// 把nums剩余的元素置零
for(int i = k; i < nums.size(); i++)
nums[i] = 0;
}
方法三(法二的优化版):
根据方法二的思想,还要把后面的元素依次赋为0,感觉不太爽。既然数组中有0,为何不在元素放在[0-k)的同时让0移到后面?
k的定义不变。遍历数组。
i 遇到 0 时,只需指针 i 向后移一。
i 遇到 非零元素 时,交换元素nums[k] 和 nums[i] ;指针 i,k 向后移一。此时区间 [ k - i ] 全为0。
优化:在未遇到第一个0前,i = k。此时只需让 i,k向后移一便可。
时间复杂度:O(n)
空间复杂度:O(1)
void moveZeroes(vector<int>& nums) {
int k = 0; //[0-k)中的元素为非零元素,从k是第一个0的位置
// 遍历数组,交换零和非零元素的位置
// 同时[k-i]为0
for(int i = 0; i < nums.size(); i++)
if( nums[i])
if( i != k )
swap( nums[k++], nums[i] );
else // 当数组中没有0的时候,就不用每一次都交换了
k++;
}