1089. 复写零

原题链接：1089. 复写零

 

Soluton:

暴力做法：

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        for(int i = 0;i < arr.size();i++) {
            if(arr[i] == 0) {
                for(int j = arr.size() - 1;j > i;j--)
                    arr[j] = arr[j - 1];
                i++;
            }           
        }
    }
};

遍历两次数组,第一遍历统计0的个数。本题有一个规律，arr[i]前有多少个0,arr[i]就得向后移动对应的位置

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int offset = 0;
        int n = arr.size();
        for(int i = 0;i < n;i++) {  //统计有多少个0
            if(arr[i] == 0) offset++;
        }

        for(int i = n - 1;i >= 0;i--) {
            if(arr[i] == 0) offset--;
            if(offset + i < n) {
                arr[i + offset] = arr[i];
                if(arr[i] == 0 && i + offset + 1 < n) {
                    arr[i + offset + 1] = 0;
                }
            }
        }
    }
};