AC日记——Card Game codeforces 808f

F - Card Game

思路：

　　题意：

　　　　有n张卡片，每张卡片三个值，pi，ci，li；

　　　　要求选出几张卡片使得pi之和大于等于给定值；

　　　　同时，任意两两ci之和不得为素数；

　　　　求选出的li的最小值，如果不能到达给定值则输出-1；

　　　　二分+网络流最小割；

代码：

#include <bits/stdc++.h>

namespace data
{
    #define maxn 105
    #define maxque 200005

    int val[maxn],mag[maxn],lev[maxn],num,lit;

    #undef maxn
    #undef maxque
}

namespace init
{
    inline void in(int &justval)
    {
        int if_z=; justval=;
        char Cget=getchar();
        while(Cget>''||Cget<'')
        {
            if(Cget=='-') if_z=-;
            Cget=getchar();
        }
        while(Cget>=''&&Cget<='')
        {
            justval=justval*+Cget-'';
            Cget=getchar();
        }
        justval*=if_z;
    }
}

namespace solve
{
    #define maxn 105
    #define maxque 200005
    #define INF 0x7fffffff

    struct EdgeType
    {
        int v,f;

        EdgeType *next,*another;

    };
    struct EdgeType *head[maxn],e[maxque];

    int que[maxque],sum,prime[maxque*],num,cnt,s,t,deep[maxn];

    bool p_[maxque*-];

    inline void edge_add(int u,int v,int f)
    {
        e[++cnt].v=v,e[cnt].f=f,e[cnt].next=head[u],head[u]=&e[cnt];
        e[++cnt].v=u,e[cnt].f=,e[cnt].next=head[v],head[v]=&e[cnt];
        e[cnt].another=&e[cnt-],e[cnt-].another=&e[cnt];
    }

    int min(const int &tops_,const int &tops__)
    {
        if(tops_<tops__) return tops_;
        else return tops__;
    }

    inline bool bfs()
    {
        t=data::num+,s=;int h=,tail=;que[h]=s;
        for(int i=s;i<=t;i++) deep[i]=-;deep[s]=;
        while(h<tail)
        {
            int now=que[h++];
            for(EdgeType *i=head[now];i!=NULL;i=i->next)
            {
                if(deep[i->v]<&&i->f)
                {
                    deep[i->v]=deep[now]+;
                    if(i->v==t) return true;
                    que[tail++]=i->v;
                }
            }
        }
        return false;
    }

    int flowing(int now,int flow)
    {
        if(now==t||flow<=) return flow;
        int oldflow=;
        for(EdgeType *i=head[now];i!=NULL;i=i->next)
        {
            if(deep[i->v]==deep[now]+&&i->f)
            {
                int pos=flowing(i->v,min(i->f,flow));
                flow-=pos,oldflow+=pos,i->f-=pos,i->another->f+=pos;
                if(!flow) return oldflow;
            }
        }
        if(!oldflow) deep[now]=-;
        return oldflow;
    }

    bool dinic(int res,int lit)
    {
        while(bfs()) res-=flowing(s,INF);
        return res>=lit;
    }

    void ouler(int limit)
    {
        for(int i=;i<=limit;i++)
        {
            if(!p_[i]) prime[++num]=i;
            for(int j=;prime[j]*i<=limit&&j<=num;j++)
            {
                p_[i*prime[j]]=true;
                if(i%prime[j]==) break;
            }
        }
    }

    bool check(int x)
    {
        int n=data::num,k=data::lit,res=;s=,t=n+,cnt=;
        for(int i=s;i<=t;i++) head[i]=NULL;
        int idx__=-;
        for(int i=;i<=n;i++)
        {
            if(data::lev[i]>x) continue;
            if(data::mag[i]==&&data::val[i]>data::val[idx__]) idx__=i;
        }
        for(int i=;i<=n;i++)
        {
            if(data::lev[i]>x||(data::mag[i]==&&idx__!=i)) continue;
            res+=data::val[i];
            if(data::mag[i]&)
            {
                edge_add(s,i,data::val[i]);
                for(int j=;j<=n;j++)
                {
                    if(data::lev[j]>x||i==j) continue;
                    if(!(data::mag[j]&))
                    {
                        if(!p_[data::mag[i]+data::mag[j]]) edge_add(i,j,INF);
                    }
                    else if(data::mag[i]+data::mag[j]==) edge_add(i,j,INF);
                }
            }
            else edge_add(i,t,data::val[i]);
        }
        return dinic(res,k);
    }

    void binary()
    {
        ouler(maxque*-);
        int l=,r=data::num,ans=-;
        while(l<=r)
        {
            int mid=l+r>>;
            if(check(mid)) ans=mid,r=mid-;
            else l=mid+;
        }
        printf("%d\n",ans);
    }

    #undef INF
    #undef maxn
    #undef maxque
}

int main()
{
    init::in(data::num),init::in(data::lit);
    for(int i=;i<=data::num;i++)
    {
        init::in(data::val[i]);
        init::in(data::mag[i]);
        init::in(data::lev[i]);
    }
    solve::binary();
    return ;
}