【LeetCode 每日一题】2460. 对数组执行操作

83 阅读2分钟

2460. 对数组执行操作

难度:简单

时间:2023/06/05


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

返回结果数组。

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

示例 1:

输入: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]

示例 2:

输入:nums = [0,1]
输出:[1,0]
解释:无法执行任何操作,只需要将 0 移动到末尾。

提示:

  • 2 <= nums.length <= 2000
  • 0 <= nums[i] <= 1000

解题思路:

先遍历一次数组,按照要求修改数组中的值,然后将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 [] res = new int[n];
        int idx = 0;
        for (int x: nums) {
            if (x != 0){
                res[idx++] = x;
            }
        }
        return res;
    }
}
​