题目
给你一个长度固定的整数数组 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]
二、思路
双指针
1,找到最后一个复写位置
定义dest=-1,cur=0;
遇到0dest跳两步,非0跳一步。
注意数组越界问题。
2.边界处理
3.从后往前复写
三、代码
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int dest=-1,cur=0, n=arr.size();
//找到最后一个复写位置
while(cur<n){
if(arr[cur]){
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]) arr[dest--] = arr[cur--];
else{
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
}
}
};