【周赛318 LeetCode 6229 】对数组执行操作

题目

给你一个下标从 0 开始的数组 nums ，数组大小为 n ，且由 非负 整数组成。

你需要对数组执行 n - 1 步操作，其中第 i 步操作（从 0 开始计数）要求对 nums 中第 i 个元素执行下述指令：

如果 nums[i] == nums[i + 1] ，则 nums[i] 的值变成原来的 2 倍，nums[i + 1] 的值变成 0 。否则，跳过这步操作。
1

在执行完 全部 操作后，将所有 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] 。
1
2
3
4
5
6
7
8
9

思路

   时间复杂度O(n) 空间复杂度O(1)
   首先遍历一次，将满足条件的进行加倍和赋0，不满足的则跳过。
然后利用双指针也就是快慢指针进行移位操作，最终慢指针left的位置就是非0的个数，记录下来，用 总长度-left 就是值为0的个数，最后将倒数的 总长度-left 个元素全赋值为0。

vector<int> applyOperations(vector<int>& nums) {
	for(int i=0;i<nums.size()-1;i++){
		if(nums[i]==nums[i+1]){
			nums[i]=2*nums[i];
			nums[i+1]=0;
			i++;
		}
	}
	//利用双指针实现 0的移位操作
	int left=0,right=0;
	for(;right<nums.size();right++){
		if(nums[right]!=0){
			nums[left++]=nums[right];
		}
	}
	int count=nums.size()-left;//记录有多少个0 后面直接将倒数的全赋值为0
	for(int j=nums.size()-1;count>0;count--)
		nums[j--]=0;

	return  nums;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21