给你一个长度固定的整数数组 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 思路简单复杂度为O^2
//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]
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int slow, fast;
slow = fast = arr.size()-1;
fast -= 1;
while (fast >=0)
{
if (arr[fast] == 0)
{
slow = arr.size()-1;
while (slow >fast)
{
arr[slow] = arr[slow - 1];
slow--;
}
//slow==fast //arr[fast]=0
fast--;
}
else
{
fast--;
}
}
}
};

代码一看就懂了,从后往前复制,就不存在越界的问题,也避免了复写零时覆盖了要移动的数
解法2
从前往后走
class Solution
{
public:
void duplicateZeros(vector<int>& arr)
{
// 1. 先找到最后⼀个数
int cur = 0, dest = -1, n = arr.size();
while (cur < n)
{
if (arr[cur]) dest++;
else dest += 2;
if (dest >= n - 1) break;
cur++;
}
// 2. 处理⼀下边界情况
if (dest == n)
{
arr[n - 1] = 0;
cur--; dest -= 2;
}
// 3. 从后向前完成复写操作
while (cur >= 0)
{
if (arr[cur]) arr[dest--] = arr[cur--];
else
{
arr[dest--] = 0;
arr[dest--] = 0;
cur--;
}
}
}
};

左边一组是没有越界的,右边是越界情况,这种思路第一次比较难想。
找最后一个数(c的任务)两种情况c都预留了两个空位,是因为在复制0时,dest(d)不仅要存储0还要存储0的下一个数
385

被折叠的 条评论
为什么被折叠?



