题目解析
给你一个长度固定的整数数组 arr ,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
注意:请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
示例 1:
输入:arr = [1,0,2,3,0,4,5,0]
输出:[1,0,0,2,3,0,0,4]
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]
示例 2:
输入:arr = [1,2,3]
输出:[1,2,3]
解释:调用函数后,输入的数组将被修改为:[1,2,3]
解题思路
在复写0的过程如果从前往后复写可能会出现没有被复写的元素被前面元素复写时给覆盖,例如在数组
arr = [1,0,2,3,0,4,5,0]中元素2被前面元素0复写的时候将元素2给覆盖了,元素2不能再被复写了,所以我们考虑从后往前复写,首先需要找到最后一个被复写 的数,定义变量cur和dest,cur遍历数组,并判断cur下标元素的值,决定dest加一还是加二,当dest大于等于arrSize-1时跳出(arrSize-1数组元素最后一个下标),此时的cur下标的值即为最后一个复写的元素,然后开始从cur位置开始从前复写。但是有一个特殊情况,例如arr[0,0,0,0,0,0,0],当curz找到最后一个复写元素时,此时的dest的值等于7(dest是从-1开始加的,相当于第0个元素的下标是-1,所以起始位置是-1),大于了数组最大下标6,越界了。这个时候将arr[arrSize-1]=0,cur–,dest-=2(arr[arrSize-1]=0,因为只有最后一个元素是0的情况才可能出现dest越界)。从倒数第二个元素从后往前开始复写即可。
代码实现
C语言实现
void duplicateZeros(int* arr, int arrSize)
{
int cur=0;
int dest=-1;
while(cur<arrSize)
{
if(arr[cur]!=0)
{
dest++;
}
else
{
dest+=2;
}
if(dest>=arrSize-1)
{
break;
}
cur++;
}
if(dest==arrSize)
{
arr[arrSize-1]=0;
cur--;
dest-=2;
}
while(cur>=0)
{
if(arr[cur]!=0)
{
arr[dest--]=arr[cur--];
}
else
{
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
}
}
Java语言实现
class Solution {
public void duplicateZeros(int[] arr)
{
int cur=0;
int dest=-1;
int n=arr.length;
while(cur<n)
{
if(arr[cur]!=0)
{
dest++;
}
else
{
dest+=2;
}
if(dest>=n-1)
{
break;
}
cur++;
}
if(dest==n)
{
arr[n-1]=0;
cur--;
dest-=2;
}
while(cur>=0)
{
if(arr[cur]!=0)
{
arr[dest--]=arr[cur--];
}
else
{
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
}
}
}
复写零题目
欢迎留言讨论