★ 算法OJ题 ★ 力扣1089 - 复写零

Ciallo～(∠・ω< )⌒☆ ~ 今天，我将和大家一起做一道双指针算法题--复写零~

目录

一  题目

二  算法解析

2.1 算法思路

2.2 算法流程

三  编写算法

---

一  题目

1089. 复写零 - 力扣（LeetCode）

二  算法解析

2.1 算法思路

此题依旧使用双指针算法，先根据异地操作，优化成双指针就地操作。

如果从前向后进⾏原地复写操作，由于 0 会复写两次，导致没有复写的数被覆盖。

因此我们选择从后往前的复写策略。但是从后向前复写的时候，我们需要找到最后⼀个复写的数，因此我们的⼤体流程分两步：

• 先找到最后⼀个复写的数。
• 从后向前进⾏复写操作。

2.2 算法流程

0.初始化 

• cur = 0, dest = -1

1. 找到最后⼀个复写的数（也使用双指针算法）

• 判断cur位置的值是否==0
• ==0 dest指针向后走两步，!=0 dest指针向后走一步
• 判断dest是否到末位置
• cur++

1.5. 判断dest是否会越界

• n-1 位置赋为0
• cur   -= 1  dest -= 2

2. 从后往前，完成复写操作

（i）判断 cur 位置的值：

• 如果是 0 ： dest 以及 dest - 1 位置修改成 0 ， dest -= 2 ；
• 如果⾮零： dest 位置修改成 0 ， dest -= 1 ；

（ii）cur-- ，复写下⼀个位置。

三  编写算法

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int cur = 0, dest = -1, n = arr.size();
        // 找到最后⼀个复写的数
        while(cur < n)
        {
            if(arr[cur] == 0)
                dest += 2;
            else
                dest++;
            if(dest >= n - 1)
                break;
            cur++;
        }
        // 判断dest边界情况
        if(dest == n)
        {
            arr[n-1] = 0;
            cur--;
            dest -= 2;
        }
        // 从后往前，完成复写操作
        while(cur >= 0)
        {
            if(arr[cur] == 0)
            {
                arr[dest] = 0;
                arr[--dest] = 0;
                dest--;
                cur--;
            }
            else
            {
                arr[dest] = arr[cur];
                dest--;
                cur--;
            }
        }
    }
};

 精简版~：

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--;
			}
		}
	}
};