【C++】双指针算法：复写零

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）。