目录
力扣1089. 复写零
难度 简单
给你一个长度固定的整数数组 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 <= arr.length <= 10^4
0 <= arr[i] <= 9
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
}
};
解析代码
从右往前的双指针问题(标的简单实际并不简单),需要先找到最后得到的vector最右边的数。这里找这个数用从左向右的双指针:
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int left = -1, right = 0, size = arr.size();
while(right < size) // 找到最后一个数
{
if(arr[right] != 0)
{
left++;
}
else
{
left += 2;
}
if(left >= size - 1)
{
break;
}
right++;
}
if(left == size) // 处理边界情况
{
arr[size - 1] = 0;
right --;
left -= 2;
}
while(right >= 0) // 从右往左复写
{
if(arr[right] != 0)
{
arr[left--] = arr[right--];
}
else
{
arr[left--] = 0;
arr[left--] = 0;
right--;
}
}
}
};