1.题目
别看这是一道简单题,它的通过率低于一些中等甚至困难的题目!
大大增加这道题目难度的是最后一句话:1.不可越界写入。2.就地修改。
如果可以再创建一个数组的话,那么这道题目就会非常简单,但这道题目必须要求在原数组上修改就需要注意很多细节!
2.算法思路
同样,我们需要两个指针:cur和dest。
cur负责遍历数组,判断所读取的元素是0还是非0。
dest负责修改数据。
现在两个指针分工明确,就需要确定是从左往右遍历数组还是从右往左遍历数组。
举个例子:arr={1,0,2,3};
先试试从左往右遍历数组:
cur读取第一个元素是非0,dest对数组不做处理。
cur读取第二个元素是0,dest将第二个,第三个元素修改成0。此时数组变成:arr={1,0,0,3};从这一步开始,后面就全错了!因为第三个元素被改成0了,cur已经读读不到原来的数据了,结果就是dest将后面的数据全部改成零!
看来只能从右往左遍历数组:
这次举题目中示例一的例子:arr={1,0,2,3,0,4,5,0};
从这个例子中结果,我们发现cur指针最多遍历到元素4就不会往后遍历了,如果我们先确定cur读取的最后一个数据的位置,让cur从那个位置从后往前遍历,让dest从后往前修改就不会出现0把原来还没读取的数据覆盖的情况。
但这种方法需要考虑一个边界情况:如果数组是arr={1,0,2,3,0,4};
修改后的结果就是arr={1,0,0,2,3,0};要特别注意dest从后往前修改数组时,最后一个0只写了一遍!把这个边界情况解决后就可以完成这道题目啦!
3.提交结果和代码
提交结果:
代码:
这个代码是我第一次写这道题目时写的代码,并不是最精简的,读起来有些费劲,我在后面给出了简化版的
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int size=arr.size(),cur=0,dest=-1;
//用cur找到数组需要复写的最后一个数
while(1){
if(size==0)
break;
if(arr[cur]){
dest++;
if(dest<size-1)
cur++;
else
break;
}
else{
dest+=2;
if(dest<size-1)
cur++;
else
break;
}
}
if(dest==size){ //处理边界情况
arr[--dest]=arr[cur--];
dest--;
}
while(cur>=0){ //dest前置--和后置--尽量不要混合使用,否则很可能在数组同一个位置会写入两次,或有的位置会漏写
if(arr[cur]==0){
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
else{
arr[dest--]=arr[cur--];
}
}
}
};
这是简化后的代码,结果更清晰一些:
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int size=arr.size(),cur=0,dest=-1;
//用cur找到数组需要复写的最后一个数
while(cur<size){
if(arr[cur]) dest++;
else dest+=2;
if(dest>=size-1) break;
cur++;
}
if(dest==size){ //处理边界情况
arr[dest-1]=arr[cur];
dest-=2;cur--;
}
while(cur>=0){ //dest前置--和后置--尽量不要混合使用,否则很可能在数组同一个位置会写入两次,或有的位置会漏写
if(arr[cur]==0){
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
else arr[dest--]=arr[cur--];
}
}
};
时间复杂度分析:
大约遍历了两次数组,时间复杂度:O(n)。
空间复杂度分析:
定义了三个变量,空间复杂度:O(1)。