问题描述
给定一个数组nums,写一个函数,将数组中的所有的0挪到数组的末尾,而维持其他所有非0元素的相对位置。
举例:num = [0, 1, 0, 3, 12],函数运行后结果为[1, 3, 12, 0, 0]
基本解法
1.C++版本
//时间复杂度:O(n)
//空间复杂度:O(n)
void moveZeroes(vector<int>& nums) {
vector<int> nonZeroElements;
for (int i = 0; i < nums.size(); i++)
nonZeroElements.push_back(nums[i]);
for (int i = 0; i < nonZeroElements.size(); i++)
nums[i] = nonZeroElements[i];
for (int i = nonZeroElements.size(); i < nums.size(); i++)
nums[i] = 0;
}
2.Java版本
public void moveZeroes(int[] nums){
ArrayList<Integer> nonZeroElements=new ArrayList<Integer>();
//将vec中所有非0元素放入nonZeroElements中
for(int i=0;i<nums.length;i++)
if(nums[i]!=0)
nonZeroElements.add(nums[i]);
//将nonZeroElements中的所有元素依次放入到nums开始的位置
for(int i=0;i<nonZeroElements.size();i++)
nums[i]=nonZeroElements.get(i);
//将nums剩余的位置放置为0
for(int i=nonZeroElements.size();i<nums.length;i++)
nums[i]=0;
}
备注:public static void main(String[] args)快捷键:psvm
优化1(简化空间复杂度)
//时间复杂度
//空间复杂度
void moveZeroes(vector<int>& nums)
int k=0;
//nums中,[0...k)的元素均为非0元素
//遍历到第i个元素后,保证[0...i]中所有fei0元素
//都按照顺序排列在[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;
}
优化2(更简洁)
void moveZeroes(vector<int>& nums){
int k=0;//nums中,[0...k)的元素均为非0元素
for(int i=0;i<nums.size();i++)
if(nums[i])
if(i!=k)//索引不等再交换
swap(nums[k++],nums[i]);
else//i==k
k++;
}
后续再改,csdn这个引入代码的操作太烂了。。。