本文属于「征服LeetCode」系列文章之一,这一系列正式开始于2021/08/12。由于LeetCode上部分题目有锁,本系列将至少持续到刷完所有无锁题之日为止;由于LeetCode还在不断地创建新题,本系列的终止日期可能是永远。在这一系列刷题文章中,我不仅会讲解多种解题思路及其优化,还会用多种编程语言实现题解,涉及到通用解法时更将归纳总结出相应的算法模板。
为了方便在PC上运行调试、分享代码文件,我还建立了相关的仓库:https://github.com/memcpy0/LeetCode-Conquest。在这一仓库中,你不仅可以看到LeetCode原题链接、题解代码、题解文章链接、同类题目归纳、通用解法总结等,还可以看到原题出现频率和相关企业等重要信息。如果有其他优选题解,还可以一同分享给他人。
由于本系列文章的内容随时可能发生更新变动,欢迎关注和收藏征服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
解法 模拟
根据题意要求,如果 n u m s [ i ] = = n u m s [ i + 1 ] nums[i] == nums[i+1] nums[i]==nums[i+1] ,需要进行以下操作:
- n u m s [ i ] = 2 × n u m s [ i ] nums[i] = 2 \times nums[i] nums[i]=2×nums[i] ;
- n u m s [ i + 1 ] = 0 nums[i + 1] = 0 nums[i+1]=0
在遍历数组执行上述操作的同时,将所有非零元素移动到数组的头部,所有 0 0 0 移动到数组的末尾。
class Solution {private void swap(int[] nums, int i, int j) {int t = nums[i]; nums[i] = nums[j]; nums[j] = t;}public int[] applyOperations(int[] nums) {int n = nums.length;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, j++); // j标志非零数组的下一个元素}return nums;}
}
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n) 。只需要遍历一遍数组即可。
- 空间复杂度: O ( 1 ) O(1) O(1) 。直接在原数组上进行修改。