【算法——双指针】LeetCode 1089 复写零

千万不要被这道题标注着“简单”迷惑了,实际上需要注意的细节很多。

题目描述:

【算法——双指针】LeetCode 1089 复写零_第1张图片

解题思路:

正序遍历,确定结果数组的最后一个元素所在的位置;知道最后一个元素的位置后倒序进行填充。

  1. 先找到最后一个需要复写的数
    1. 先判断cur位置的值
    2. 决定dest向后移动一步还是两步
    3. 判断dest是否已经到结束位置
    4. cur++
  2. 处理边界情况(dest超过arr.size()-1的情况) 
  3. 从后向前完成复写的操作

代码:

class Solution {
public:
    void duplicateZeros(vector& arr) {
        int cur = 0;
        int dest = -1;
        while(cur < arr.size())
        {
            if(arr[cur] == 0)
            {
                dest += 2;
            }
            else
            {
                dest++;
            }
            if(dest >= arr.size() - 1)
            {
                break;
            }
            cur++;
        }
        if(dest > arr.size() - 1)
        {
            arr[arr.size() - 1] = 0;
            cur--;
            dest -= 2;
        }
        while(cur >= 0)
        {
            if(arr[cur] != 0)
            {
                arr[dest] = arr[cur];
                dest--;
                cur--;
            }
            else
            {
                arr[dest] = 0;
                arr[dest - 1] = 0;
                cur--;
                dest -= 2;
            }
        }
    }
};

结果:

【算法——双指针】LeetCode 1089 复写零_第2张图片

你可能感兴趣的:(刷题,leetcode,数据结构,算法)