【每日一题Day228】LC2460对数组执行操作 | 模拟+双指针

原创已于 2023-06-12 10:32:01 修改 · 602 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #数据结构

于 2023-06-05 13:56:24 首次发布

每日一题同时被 2 个专栏收录

371 篇文章

订阅专栏

模拟

57 篇文章

订阅专栏

该问题要求对一个非负整数数组执行特定操作，每次比较相邻元素，若相等则加倍第一个并置第二个为0。最后，通过双指针方法将非0元素移到数组前端，0移到末尾。给定的解决方案实现了这一过程，具有线性时间复杂度和常量空间复杂度。

对数组执行操作【LC2460】

给你一个下标从 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] 。

返回结果数组。

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

思路

简单模拟题，执行完所有操作后，使用双指针将非0的元素移动数组前端，同时将末尾元素赋值为0

实现

class Solution {
    public int[] applyOperations(int[] nums) {
        int n = nums.length;
        for (int i = 0; i < n - 1; i++){
            if (nums[i] == nums[i + 1]){
                nums[i] *= 2;
                nums[i + 1] = 0;
            }

        }
        int i = 0, j = 0;
        while(j < n){
            nums[i] = nums[j];
            if (i != j){
                nums[j] = 0;
            }
            if (nums[i] != 0){// 不为0移动指针；为0不移动
                i++;
            }
            j++;                  
        }
        return nums;
    }
}