給你一個下標從 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
同向雙指針,右指針負責執行題目操作,左指針負責將所有非0值都放到左邊:
class Solution {
public:vector<int> applyOperations(vector<int>& nums) {int l = 0;int r = 0;int n = nums.size();while (r < n - 1) {if (nums[r] == nums[r + 1]) {nums[r] *= 2;nums[r + 1] = 0;}if (nums[r] != 0) {swap(nums[l], nums[r]);++l;}++r;}if (nums[r] != 0) {swap(nums[l], nums[r]);}return nums;}
};
如果nums的長度為n,則此算法時間復雜度為O(n),空間復雜度為O(1)。