cf721 D. Maxim and Array(贪心)

题意:

对给定数组(可能有负数)进行最多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;
}

上一篇:JavaScript-100:作用域链


下一篇:11.18训练赛