2460. 对数组执行操作

努力向上的开拓，才使弯曲的竹鞭化作了笔直的毛竹。

LC每日一题，参考2460. 对数组执行操作。

题目

给你一个下标从 0 开始的数组 nums ，数组大小为 n ，且由 非负 整数组成。

你需要对数组执行 n - 1 步操作，其中第 i 步操作（从 0 开始计数）要求对 nums 中第 i 个元素执行下述指令：

• 如果 nums[i] == nums[i + 1] ，则 nums[i] 的值变成原来的 2 倍，nums[i + 1] 的值变成 0 。否则，跳过这步操作。

在执行完 全部 操作后，将所有 0 移动 到数组的 末尾 。

• 例如，数组 [1,0,2,0,0,1] 将所有 0 移动到末尾后变为 [1,2,1,0,0,0] 。

返回结果数组。

注意 操作应当 依次有序 执行，而不是一次性全部执行。

输入：nums = [1,2,2,1,1,0]
输出：[1,4,2,0,0,0]
解释：执行以下操作：
- i = 0: nums[0] 和 nums[1] 不相等，跳过这步操作。
- i = 1: nums[1] 和 nums[2] 相等，nums[1] 的值变成原来的 2 倍，nums[2] 的值变成 0 。数组变成 [1,4,0,1,1,0] 。
- i = 2: nums[2] 和 nums[3] 不相等，所以跳过这步操作。
- i = 3: nums[3] 和 nums[4] 相等，nums[3] 的值变成原来的 2 倍，nums[4] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。
- i = 4: nums[4] 和 nums[5] 相等，nums[4] 的值变成原来的 2 倍，nums[5] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。
执行完所有操作后，将 0 全部移动到数组末尾，得到结果数组 [1,4,2,0,0,0] 。

解题思路

• 滑动窗口：分两步,先按指令处理数组,再用滑动窗口移动0到后面。
• 模拟：指令处理后,可以按下标顺序对数组重新赋值。

滑动窗口

class Solution {
    public int[] applyOperations(int[] nums) {
        for(int i = 0; i < nums.length-1; i++) {
            if(nums[i] == nums[i+1]) {
                nums[i] *= 2;
                nums[i+1] = 0;
            }
        }
        //滑动窗口
        for(int i = 0,j = -1;i < nums.length; i++) {
            if(j == -1) {
                if(nums[i] == 0) j = i;//记录0位置
            }else if(nums[i] != 0 && j < nums.length) {
                nums[j] = nums[i];
                nums[i] = 0;
                //计算下一个0位置
                j++;
                while(j < nums.length && nums[j] != 0)j++;
            }
        }
        return nums;
    }
}

复杂度分析

• 时间复杂度：O(n)，遍历次数不超过3n,n为数组长度。
• 空间复杂度：O(1)。

模拟

class Solution {
    public int[] applyOperations(int[] nums) {
        for(int i = 0,id = 0; i < nums.length; i++) {
            if(i+1 < nums.length && nums[i] == nums[i+1]) {
                nums[i] *= 2;
                nums[i+1] = 0;
            }
            if(nums[i] != 0){
                nums[id] = nums[i];
                if(id < i)nums[i] = 0;
                id++;
            } 
        }
        return nums;
    }
}

复杂度分析

• 时间复杂度：O(n)，总共遍历数组一次。
• 空间复杂度：O(1)。