【LetMeFly】2460.对数组执行操作

力扣题目链接：https://leetcode.cn/problems/apply-operations-to-an-array/

给你一个下标从 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，第二个元素二倍。

接着关于将非零元素移动到数组前：

• 对于可变数组（如C++），我们可以使用一个指针指向非零元素填充到的位置，将剩余位置全部置0
• 对于不可变数组（如Python），我们可以直接新开辟一个数组并将非零元素放入数组前方，其余元素用0补齐

End.

• 时间复杂度$O(len(nums))$
• 空间复杂度：对于可变数组的编程语言$O(1)$，对于不可变数组的编程语言$O(len(nums))$

AC代码

C++

class Solution {
    
    
public:
    vector<int> applyOperations(vector<int>& nums) {
    
    
        for (int i = 0; i + 1 < nums.size(); i++) {
    
    
            if (nums[i] == nums[i + 1]) {
    
    
                nums[i] *= 2;
                nums[i + 1] = 0;
            }
        }
        int loc = 0;
        for (int i = 0; i < nums.size(); i++) {
    
    
            if (nums[i]) {
    
    
                nums[loc++] = nums[i];
            }
        }
        while (loc < nums.size()) {
    
    
            nums[loc++] = 0;
        }
        return nums;
    }
};

Python

# from typing import List

class Solution:
    def applyOperations(self, nums: List[int]) -> List[int]:
        for i in range(0, len(nums) - 1):
            if nums[i] == nums[i + 1]:
                nums[i] *= 2
                nums[i + 1] = 0
        ans = []
        for i in nums:
            if i:
                ans.append(i)
        ans += [0] * (len(nums) - len(ans))
        return ans

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/131046033