【思维构造】Find The Array—CF1463B

Find The Array—CF1463B

思路

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

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

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

$Code$

#include 
#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;
}