对数组执行操作【LC2460】
给你一个下标从 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]
。
返回结果数组。
注意 操作应当 依次有序 执行,而不是一次性全部执行。
- 思路
简单模拟题,执行完所有操作后,使用双指针将非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 i = 0, j = 0; while(j < n){ nums[i] = nums[j]; if (i != j){ nums[j] = 0; } if (nums[i] != 0){// 不为0移动指针;为0不移动 i++; } j++; } return nums; } }