目录
题目
思路
代码
题目
给你一个长度固定的整数数组 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]
思路
题目依然是对数组元素原地进行修改,所以可以采用双指针的思路来解题,一个负责遍历一个负责更改,所以可以设立两个指针,cur和dest,cur负责遍历数组,dest负责更改数据,如果非零,dest就复写一遍,如果是0,dest就往后写入两个0。
可通过观察发现,如果从前往后进行遍历更改,就会导致数据被覆盖。从而破坏了数组的完整性。
因为在对数组进行修改时,多余的元素会从数组的队尾出,所以可以从后往前去更改数组。
所以我们可以先去找到最后一个需要复写的数,这样就可以确定数组的最后一个元素的值,从而可
以更好的控制写入时不发生溢出。
如上所示,当dest走到头时此时cur的位置就是最后一个需要复写的数。
但当以下情况出现时,dest会出现越界的情况,如果直接去访问就会导致越界问题。
所以当碰到越界以后需要进行一个特殊处理,当dest==n时说明越界了,也说明最后一个复写的元素绝对是0才会导致dest越界,所以直接把下标为n-1也就是数组最后一个元素的值修改为0然后让dest-=2,同时让cur往前移动一位。此时第一步就完成了。接下来就可以修改数组数据了。
先判断cur对应的元素是否为0,如果不为0 dest修改该位置元素然后向前移动一位,如果为0 将该位置元素改为0然后向前把前一个位置元素也改为0。
代码
class Solution {
public:
void duplicateZeros(vector<int>& arr)
{
int cur=0,dest=-1,n=arr.size();
while(cur<n)//找最后一个需要复写的下标
{
if(arr[cur]) ++dest;
else dest+=2;
if(dest>=n-1) break;
cur++;
}
if(dest==n)
{
dest-=2;
--cur;
arr[n-1]=0;
}
while(cur>=0)
{
if(arr[cur])arr[dest--]=arr[cur];
else
{
arr[dest--]=0;
arr[dest--]=0;
}
cur--;
}
}
};