[PA2014]Bazarek

[PA2014]Bazarek

题目大意：

有\(n(n\le10^6)\)件商品，\(m(m\le10^6)\)次询问。每次询问若选出其中的\(k\)个，要求它们的总价为奇数，求最大可能的总价。

思路：

从大到小排序取前\(k\)个，若不是奇数就去掉已选最小偶数/奇数再加上未选最大奇数/偶数。

源代码：

#include<cstdio>
#include<cctype>
#include<algorithm>
#include<functional>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
typedef long long int64;
const int N=1e6+1;
int64 a[N],min[N][2],max[N][2];
int main() {
    const int n=getint();
    for(register int i=1;i<=n;i++) {
        a[i]=getint();
    }
    std::sort(&a[1],&a[n]+1,std::greater<int64>());
    for(register int i=1;i<=n;i++) {
        if(i!=1) {
            min[i][0]=min[i-1][0];
            min[i][1]=min[i-1][1];
        }
        min[i][a[i]&1]=a[i];
    }
    for(register int i=n;i>=1;i--) {
        if(i!=n) {
            max[i][0]=max[i+1][0];
            max[i][1]=max[i+1][1];
        }
        max[i][a[i]&1]=a[i];
    }
    for(register int i=1;i<=n;i++) {
        a[i]+=a[i-1];
    }
    const int m=getint();
    for(register int i=0;i<m;i++) {
        const int k=getint();
        if(a[k]&1) {
            printf("%lld\n",a[k]);
            continue;
        }
        int64 ans=0;
        if(k!=n&&min[k][0]&&max[k+1][1]) {
            ans=std::max(ans,a[k]-min[k][0]+max[k+1][1]);
        }
        if(k!=n&&min[k][1]&&max[k+1][0]) {
            ans=std::max(ans,a[k]-min[k][1]+max[k+1][0]);
        }
        printf("%lld\n",ans?:-1);
    }
    return 0;
}