复写0
1089. 复写零 - 力扣(LeetCode)
给你一个长度固定的整数数组 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 <= 104
0 <= arr[i] <= 9
思路如图:
Java代码:
class Solution {
public void duplicateZeros(int[] arr) {
int cur = 0 , dest = -1, n = arr.length;
// 1. 找到最后复写位置的数
while(cur < n)
{
if(arr[cur] == 0) dest += 2;
else dest += 1;
if(dest >= n - 1) break;
cur ++;
}
// 2. 处理边界问题
if(dest == n)
{
arr[n - 1] = 0;
cur--;
dest -= 2;
}
// 3. 从后向前复写
while(cur >= 0)
{
if(arr[cur] != 0) arr[dest--] = arr[cur--];
else{
arr[dest--] = 0;
arr[dest--] = 0;
cur--;
}
}
}
}
C++代码:
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int n = arr.size(), cur = 0, dest = -1;
while(cur < n)
{
if(arr[cur] == 0) dest += 2;
else dest += 1;
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 -= 1;
}
}
}
};