class Solution {
public:
vector<int> applyOperations(vector<int>& nums) {
}
};
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] 。返回结果数组。
注意 操作应当 依次有序 执行，而不是一次性全部执行。
示例 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 <= 20000 <= nums[i] <= 1000原站题解
java 解法, 执行用时: 0 ms, 内存消耗: 42.4 MB, 提交时间: 2023-06-05 09:34:28
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] <<= 1;
nums[i + 1] = 0;
}
}
int[] ans = new int[n];
int i = 0;
for (int x : nums) {
if (x > 0) {
ans[i++] = x;
}
}
return ans;
}
}
golang 解法, 执行用时: 0 ms, 内存消耗: 2.8 MB, 提交时间: 2023-06-05 09:33:56
// 模拟
func applyOperations(nums []int) []int {
n := len(nums)
for i := 0; i < n-1; i++ {
if nums[i] == nums[i+1] {
nums[i] <<= 1
nums[i+1] = 0
}
}
ans := make([]int, n)
i := 0
for _, x := range nums {
if x > 0 {
ans[i] = x
i++
}
}
return ans
}
python3 解法, 执行用时: 40 ms, 内存消耗: 15.1 MB, 提交时间: 2022-11-09 10:22:17
class Solution:
def applyOperations(self, nums: List[int]) -> List[int]:
n = len(nums)
for idx in range(n-1):
if nums[idx] == nums[idx+1] and nums[idx] > 0:
nums[idx], nums[idx+1]= nums[idx]*2, 0
return sorted(nums, key=lambda item: 0 if item > 0 else 1)
python3 解法, 执行用时: 40 ms, 内存消耗: 15 MB, 提交时间: 2022-11-09 10:19:45
class Solution:
def applyOperations(self, nums: List[int]) -> List[int]:
idx = 0
while idx < len(nums) - 1:
if nums[idx] == nums[idx+1] and nums[idx] > 0:
nums[idx], nums[idx+1]= nums[idx]*2, 0
idx += 1
idx += 1
return sorted(nums, key=lambda item: 0 if item > 0 else 1)