力扣 2460. 对数组执行操作

题目描述：2460. 对数组执行操作

给你一个下标从О开始的数组nums ，数组大小为n，且由非负整数组成。
你需要对数组执行n - 1步操作，其中第i步操作（从О开始计数)要求对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]。
返回结果数组。
注意操作应当依次有序执行，而不是一次性全部执行。

 示例：

代码：

//给你一个下标从О开始的数组nums ，数组大小为n，且由非负整数组成。你需要对数组执行n -1步操作，其中第i步操作(从О开始计数)要求对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].
//返回结果数组。
//注意操作应当依次有序执行，而不是一次性全部执行。
#include<iostream>
#include<vector>
using namespace std;

class Solution {
public:
	vector<int> applyOperations(vector<int>& nums) {
		int i = 0, j = 0;
		for (i = 0; i < nums.size()-1; i++) {
			if (nums[i] == nums[i + 1]) {
				nums[i] = nums[i] * 2;
				nums[i + 1] = 0;
			}
			if (nums[i] != 0) {
				swap(nums[i], nums[j]);
				j++;
			}
		}
		if (nums[i] != 0) {
			swap(nums[i], nums[j]);
		}
		return nums;
	}
};

int main() {
	vector<int> nums;
	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int temp;
		cin >> temp;
		nums.push_back(temp);
	}
	vector<int> res;
	Solution solution;
	res = solution.applyOperations(nums);
	for (int i = 0; i < res.size(); i++) {
		cout << res[i] << ' ';
	}
	return 0;
}

 技巧：把0移动到数组末尾可以等价与把非零移动到数组前方