JZOJ 5274. 数组

Description

Input

Output

Sample Input

输入样例1：

3 2 7
5 4 2

输入样例2：

5 3 1
5 4 3 5 5

Sample Output

输出样例1：

999999732

输出样例2：

0

Data Constraint

Solution

• 显然，乘积最好是负数，这样会更优，那么用一个布尔型变量 $Pd$ 表示当前的负数数量的奇偶性。

• 因为加减操作的对象的值的绝对值一定是当前最小的，这样才会显得更优（贪心）。

• 于是我们维护一个堆，取出其绝对值最小的那个，再对 $Pd$ 分类讨论：

  1. 若负数数量为奇数，说明乘积为负数；

  2. 若负数数量为偶数，说明乘积为正数；

• 再根据当前值的正负性来判断加减，之后压回堆中即可，时间复杂度 $O(K\ log\ N)$ 。

Code

#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
typedef long long LL;
const int N=2e5+1,mo=1e9+7;
int n,k,x;
int a[N];
bool pd;
struct data
{
    int id;
    data(int x){id=x;}
    friend bool operator <(data x,data y)
    {
        return abs(a[x.id])>abs(a[y.id]);
    }
};
priority_queue<data>q;
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
int main()
{
    n=read(),k=read(),x=read();
    for(int i=1;i<=n;i++)
    {
        a[i]=read();
        if(a[i]<0) pd=!pd;
        q.push(data(i));
    }
    while(k--)
    {
        int t=q.top().id;
        q.pop();
        if(a[t]<0)
        {
            if(pd) a[t]-=x; else a[t]+=x;
            if(a[t]>=0) pd=!pd;
        }else
        {
            if(pd) a[t]+=x; else a[t]-=x;
            if(a[t]<0) pd=!pd;
        }
        q.push(t);
    }
    LL ans=1;
    for(int i=1;i<=n;i++) ans=ans*a[i]%mo;
    printf("%lld",(ans+mo)%mo);
    return 0;
}