【2460. 对数组执行操作】

最新推荐文章于 2024-10-27 18:12:53 发布

千北@

最新推荐文章于 2024-10-27 18:12:53 发布

阅读量390

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/Sugar_wolf/article/details/131042271

版权

LeetCode 专栏收录该内容

397 篇文章 3 订阅

订阅专栏

给定一个非负整数数组，按照指定的操作进行数组元素的翻倍和置零，然后将所有0移动到数组末尾。文章提供了一个直接模拟操作的C++解决方案，其时间复杂度为O(n)，空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

来源：力扣（LeetCode）

描述：

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

方法：直接模拟

思路与算法

根据题意要求，如果 nums[i] == nums[i+1] ，则需要进行以下变换：

nums[i] = 2 × nums[i]；
nums[i+1] = 0；

在执行上述完操作后，将所有 0 移动到数组的末尾。在遍历数组时，遇到前后相等的两个元素时直接进行变换，在遍历的同时进行原地交换，将所有非零的元素移动到数组的头部。

代码：

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

执行用时：0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗：8.6 MB, 在所有 C++ 提交中击败了81.28%的用户
复杂度分析
时间复杂度：O(n)，其中 n 表示数组的长度。只需遍历一遍数组即可，因此时间复杂度为 O(n)。
空间复杂度：O(1)。直接在原数组上进行修改即可，不需要占用额外的空间。
author：LeetCode-Solution