【思维构造】Find The Array—CF1463B

Find The Array---CF1463B

思路

我太菜了，这么简单的思路都没有想到。。。

分别计算 a a a 数组中下标为奇数的所有元素的和 sum_odd 和下标为偶数的所有元素的和 sum_even。

如果 sum_odd < sum_even，将 a a a 中的所有下标为奇数的元素全部换成 1 1 1 后的 a a a 就是需要求的 b b b 数组；

如果 sum_odd > sum_even，将 a a a 中的所有下标为偶数的元素全部换成 1 1 1 后的 a a a 就是需要求的 b b b 数组；

C o d e Code Code

#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n;
int a[N];

void solve() {
	cin >> n;
	for (int i = 1; i <= n; i ++) cin >> a[i];
	
	int sum_odd = 0, sum_even = 0;
	for (int i = 1; i <= n; i ++) {
		if (i % 2) {
			sum_odd += a[i];
		} else {
			sum_even += a[i];
		}
	}
	
	for (int i = 1 + (sum_even < sum_odd); i <= n; i += 2) {
		a[i] = 1;
	}
	for (int i = 1; i <= n; i ++) {
		cout << a[i] << ' ';
	}
	cout << "\n";
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int T = 1;
	cin >> T; cin.get();
	while (T --) solve();
	return 0;
}