bzoj 3721: PA2014 Final Bazarek 贪心

如果没有限制，直接取前 $k$ 大即可.

有限制，则只有几种可能：奇换偶，偶换奇.

维护奇数偶数的前缀最小值和后缀最大值即可.

code:

#include <bits/stdc++.h>
#define N 1000005
#define LL long long
#define inf 2000000000
using namespace std;
namespace IO
{
    char *p1,*p2,buf[100000];
    #define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
    int rd() {int x=0; char c=nc(); while(c<48) c=nc(); while(c>47) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x;}
};
void setIO(string s)
{
    string in=s+".in";
    freopen(in.c_str(),"r",stdin);
}
LL a[N],f0[N],f1[N],g0[N],g1[N],sum[N];
bool cmp(LL a,LL b) { return a>b; }
int main()
{
    // setIO("input");
    int n,i,j,m;
    scanf("%d",&n);
    for(i=1;i<=n;++i) a[i]=(LL)IO::rd();
    sort(a+1,a+1+n,cmp);
    f0[0]=f1[0]=inf;
    for(i=1;i<=n;++i)
    {
        sum[i]=sum[i-1]+a[i];
        f0[i]=f0[i-1],f1[i]=f1[i-1];
        if(a[i]%2==0) f0[i]=a[i];
        else f1[i]=a[i];
    }
    g0[n]=g1[n]=-inf;
    for(i=n-1;i>=1;--i)
    {
        g0[i]=g0[i+1],g1[i]=g1[i+1];
        if(a[i+1]%2==0) g0[i]=a[i+1];
        else g1[i]=a[i+1];
    }
    m=IO::rd();
    for(i=1;i<=m;++i)
    {
        int k;
        k=IO::rd();
        if(sum[k]&1) printf("%lld\n",sum[k]);
        else
        {
            LL ans=0ll;
            if(f0[k]!=inf&&g1[k]!=-inf) ans=max(ans,sum[k]-f0[k]+g1[k]);
            if(f1[k]!=inf&&g0[k]!=-inf) ans=max(ans,sum[k]-f1[k]+g0[k]);
            if(!ans) printf("-1\n");
            else printf("%lld\n",ans);
        }
    }
    return 0;
}