bzoj 3721: PA2014 Final Bazarek 单调栈+贪心

来源：互联网发布：阿里云ai 编辑：程序博客网时间：2024/05/22 12:56

题意

有n件商品，选出其中的k个，要求它们的总价为奇数，求最大可能的总价。
n,m<=1000000,1<=价格<=10^9

分析

考虑如何从选k个数的状态转移到选k+1个数的状态。可以假如一个偶数，也可以假如两个奇数同时在原来已选的数中删掉一个偶数。而如果这两种情况都不满足，则选k+1个物品的情况无解。
我们可以用四个单调栈来维护这个过程。
一开始WA是因为没有想到若k+1不行的话不代表k+2不行。我们可以通过加入两个奇数来转移到k+2的状态。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;typedef long long LL;const int N=1000005;int a1[N],a2[N],b1[N],b2[N],top1,top2,top3,top4,n,m;LL ans[N];int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int main(){    n=read();    for (int i=1;i<=n;i++)    {        int x=read();        if (x&1) a1[++top1]=x;        else b1[++top2]=x;    }    sort(a1+1,a1+top1+1);sort(b1+1,b1+top2+1);    if (!top1) top2=0;    a2[++top3]=a1[top1];top1--;    ans[1]=a2[top3];    for (int i=2;i<=n;i++)    {        if ((top1<2||!top4)&&!top2)        {            if (top1<2) break;            i++;            ans[i]=ans[i-2]+a1[top1]+a1[top1-1];            a2[++top3]=a1[top1];top1--;a2[++top3]=a1[top1];top1--;            continue;        }        if (!top2)        {            ans[i]=ans[i-1]+a1[top1]+a1[top1-1]-b2[top4];            b1[++top2]=b2[top4];top4--;            a2[++top3]=a1[top1];top1--;a2[++top3]=a1[top1];top1--;        }        else if (top1<2||!top4)        {            ans[i]=ans[i-1]+b1[top2];            b2[++top4]=b1[top2];top2--;        }        else        {            LL w1=a1[top1]+a1[top1-1]-b2[top4],w2=b1[top2];            if (w1>w2) ans[i]=ans[i-1]+w1,a2[++top3]=a1[top1],top1--,a2[++top3]=a1[top1],top1--,b1[++top2]=b2[top4],top4--;            else ans[i]=ans[i-1]+w2,b2[++top4]=b1[top2],top2--;        }    }    m=read();    while (m--)    {        int x=read();        if (!ans[x]) puts("-1");        else printf("%lld\n",ans[x]);    }    return 0;}

阅读全文

0 0