Vupsen, Pupsen and 0
题意:
给一个数组a,找出一个数组b,使为0;
思路:
这道题也没怎么细想,就纯暴力去做了。
首先用map标记数组中元素,前后遇到的先凑成一对,最后把没凑成对的全部统计下来,最后这些元素凡是跟和不相等的都处理出相同的就好了,但f位置要减一下和,以此保证最后和为0;
#include <bits/stdc++.h>
void solve() {
int n;
std::cin >> n;
std::vector<int> a(n);
std::unordered_map<int, int>p;
std::vector<bool> st(n);
std::vector<int> ans(n);
for (int i = 0; i < n; i++) {
std::cin >> a[i];
st[i] = false;
}
int tt = 0;
for (int i = 0; i < n; i++) {
if (!p[a[i]]) {
p[a[i]] = i;
} else if (p[a[i]] && tt < n / 2 - 1) {
ans[p[a[i]]] = 1, ans[i] = -1;
st[p[a[i]]] = true, st[i] = true;
p[a[i]] = 0;
tt++;
}
}
int sum = 0;
for (int i = 0; i < n; i++) {
if (!st[i]) {
sum += a[i];
}
}
int f = -1;
for (int i = 0; i < n; i++) {
if (!st[i] && a[i] != sum) {
f = i;
}
}
for (int i = 0; i < n; i++) {
if (!st[i]) {
ans[i] = a[f];
}
}
for (int i = 0; i < n; i++) {
std::cout << ((i == f) ? (a[i] - sum) : ans[i]) << " \n"[i == n - 1];
}
}
int main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int t;
std::cin >> t;
while (t--) {
solve();
}
return 0;
}
但后来发现使想太麻烦了,跟队里大佬交流后发现有更简单的方法。
结论很好推。
AC代码如下:
#include <bits/stdc++.h>
using namespace std;
int n;
const int N = 1e5 + 5;
long long a[N];
int main() {
int t;
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%lld", &a[i]);
if (n & 1) {
if (a[1] + a[2])
cout << -a[3] << ' ' << -a[3] << ' ' << (a[1] + a[2]) << ' ';
else if (a[1] + a[3])
cout << -a[2] << ' ' << (a[1] + a[3]) << ' ' << -a[2] << ' ';
else
cout << (a[2] + a[3]) << ' ' << -a[1] << ' ' << -a[1] << ' ';
} else
cout << -a[2] << ' ' << a[1] << ' ';
for (int i = (n & 1 ? 4 : 3); i <= n; i += 2)
cout << -a[i + 1] << ' ' << a[i] << ' ';
cout << endl;
}
return 0;
}
但时间上还是有点差距的。