来源:力扣(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