题意:
对给定数组(可能有负数)进行最多k次操作,每次让一个数 +x 或 -x。要使最终所有数的乘积最小,求操作方案。
思路:
如果负号的数量为偶,就尽量把绝对值最小的数的符号改变。(改变后负号数为奇)
如果负号数为奇,就让当前绝对值最小的数的绝对值增加 x
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using PLI = pair<ll, int>;
const int N = 2e5 + 5;
ll n, k, x, a[N]; bool fu;
priority_queue<PLI, vector<PLI>, greater<PLI>> qu; //小根堆
signed main()
{
scanf("%lld%lld%lld", &n, &k, &x);
for(int i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
qu.push({abs(a[i]), i});
if(a[i] < 0) fu ^= 1;
}
while(k--)
{
ll v = qu.top().first, id = qu.top().second;
if(!fu) {
if(a[id] >= 0) a[id] -= x, fu = a[id] < 0;
else a[id] += x, fu = a[id] >= 0;
}
else {
if(a[id] >= 0) a[id] += x;
else a[id] -= x;
}
qu.pop(), qu.push({abs(a[id]), id});
}
for(int i = 1; i <= n; i++) printf("%lld ", a[i]);
return 0;
}