leetcode原题——复写0

复写0

给你一个长度固定的整数数组 arr，请你将该数组中出现的每个零都复写一遍，并将其余的元素向右平移。
注意：请不要在超过该数组长度的位置写入元素。

示例 1：
输入：[1,0,2,3,0,4,5,0]
输出：null
解释：调用函数后，输入的数组将被修改为：[1,0,0,2,3,0,0,4]
示例 2：
输入：[1,2,3]
输出：null
解释：调用函数后，输入的数组将被修改为：[1,2,3]

既然是要复写0，那么解题步骤就是这么几步

1. 找到0所在位置
2. 将其余元素包括0所在位置向后移动，将0复写一位

那么向后移动，那就应该从后往前遍历，避免覆盖元素

那么直接写出代码

void duplicateZeros(int* arr, int arrSize)
{
    for(int i=0;i<arrSize;i++)
    {
        if(arr[i]==0)
        {
            //从后往前遍历,将数组元素向右平移,覆盖元素
            for(int j=arrSize-1;j>i;j--)
            {
                arr[j]=arr[j-1];                
            }
        }
    }
}

哎？为什么后面全部都是0呢？于是画个图仔细看下便发现

每次复写0完后,如果i这个指向一次只改变一个,那将会陷入0的无限覆盖中。所以应该每次覆盖完都让i的向后指一个

void duplicateZeros(int* arr, int arrSize)
{
    for(int i=0;i<arrSize;i++)
    {
        if(arr[i]==0)
        {
            //从后往前遍历,将数组元素向右平移,覆盖元素
            for(int j=arrSize-1;j>i;j--)
            {
                arr[j]=arr[j-1];                
            }
            i++;//很关键的i++,因为复写0,0后面还有一个0,所以必须两次i++跳出0,否则将全是0
        }
    }
}

最后得以通过

这题之前困惑了我好一段时间，现在想明白了，赶紧写篇文章分享一下思路