题目
给定一个数组
nums
,编写一个函数将所有0
移动到数组的末尾,同时保持非零元素的相对顺序。请注意 ,必须在不复制数组的情况下原地对数组进行操作。
分析
在本题我一开始选择遍历数组的时候,让元素0形成一串0,在遇到非0的元素时将0串的头位置和该元素互换值,让0串向数组末尾走。
对于如何让元素0形成一串0很简单。给定一个数组 nums,设标记为 i。
如果nums[i] == 0,标记判断下一个元素,并且记录0串的长度(n)加一;如果nums[i] != 0,0串头位置和nums[i]互换值,即:
int temp = nums[i]; nums[i] = nums[i-n]; nums[i-n] = temp;
上面的代码仅解释nums[i] != 0的情况,关于头位置为什么是 i-n,自己画图就一目了然了。
这样我们就可以写出这题的代码了:
void moveZeroes(int* nums, int numsSize) {
int n = 0;
for(int i = 0; i < numsSize; i++){
if(nums[i] == 0){
n = n+1;
}
else{
int temp = nums[i];
nums[i] = nums[i-n];
nums[i-n] = temp;
}
}
}
然后附上运行结果:
其实写完之后我就在想:如果数组是0和非0元素相间分布,这种解法花的时间是很久的,是否有一种更优的解法来避免这个问题?
在翻看其他大佬的题解后我才恍然大悟。
原来只需要将非0元素按顺序放到数组前面,后面的元素全部置0就可以了。
原因很简单:题目对排序后的数组要求有个很关键的地方——保持非零元素的相对顺序,也就是要求非0元素在数组中的出现顺序不变。
代码:
void moveZeroes(int* nums, int numsSize) {
int n = 0;
for(int i = 0; i < numsSize; i++){
if(nums[i] != 0){
nums[n] = nums[i];
n = n +1;
}
}
for(int i = n; i < numsSize; i++){
nums[i] = 0;
}
}
运行结果:
总结
在做这道题的时候我一直在对元素0进行操作,反而忽略了题干中说的非0元素。有时候解题可以试着转换操作对象,说不定会更加快捷。