LeetCode 2460.对数组执行操作
题目:
给你一个下标从 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]
。
返回结果数组。
注意 操作应当 依次有序 执行,而不是一次性全部执行。
思路:移动非零,后面全零
代码:
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 left = 0;for (int j = 0; j < n; j++) {if (nums[j] != 0) {nums[left++] = nums[j];}}for (int k = left; k < n; k++) {nums[k] = 0;}return nums;}
}
性能:
时间复杂度 o(n)
空间复杂度 o(1)