Codeforces Global Round 11 A. Avoiding Zero（前缀和）

题目链接：https://codeforces.com/contest/1427/problem/A

题意

将 (n) 个数重新排列使得不存在为 (0) 的前缀和。

题解

计算正、负前缀和，如果二者和为 (0)，则不存在满足题意的排列，否则将绝对值较大的一方排在前面即可。

代码

#include <bits/stdc++.h>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int t;
	cin >> t;
	while (t--) {
		int n;
		cin >> n;
		vector<int> a(n);
		int pos = 0, neg = 0;
		for (auto &x : a) {
			cin >> x;
			(x < 0 ? neg : pos) += x;
		}
		if (pos + neg == 0) {
			cout << "NO" << "
";
			continue;
		}
		if (pos + neg > 0) {
			sort(a.begin(), a.end(), greater<>());
		} else {
			sort(a.begin(), a.end(), less<>());
		}
		cout << "YES" << "
";
		for (int i = 0; i < n; i++) {
			cout << a[i] << " 
"[i == n - 1];
		}
	}
	return 0;
}