cf 730i

题意:有n个人,每个人有两个能力值,选a个人用它的第一个能力值,b个人用它的第二个能力值,每个人只能选一次,求一个方案使得能力值之和最大,并输出选择方案.

题解:最小费用最大流,原点1向n个人每个人i+1连流量为1费用为0的边,在有两个点n+2和n+3,每个人i+1向点n+2连流量为1,费用为-能量值的边,向n+3连流量为1,费用为-能力值的边,n+2向汇点n+4连流量为a费用为0的边,n+3向汇点n+4连流量为b费用为0的边.

注意:应该在跑完了费用流之后再输出路径,因为在过程之后总有可能有反向边,所以无法在过程中就确定最终路径.跑完了费用流之后,如果某一条边反向流不为0即表示有流量经过.

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<climits>
 #include<queue>
 #include<cmath>
 #define MAXP 3010
 #define MAXE 40500
 #define MIN(a,b) (a<b?a:b)
 using namespace std;
 int n,m;
 struct Edge
 {
     int s,t,f,c,next;
 };
 class MCMF
 {
 public:
     MCMF(int s,int t);
     void insert(int s,int t,int f,int c);
     bool spfa();
     void mcmf();
     void print();
 private:
     bool *isq;
     int s,t;
     int *dist;
     int ent;
     int cost,flow;
     int *head;
     int *pre;
     Edge *edge;
 };
 MCMF::MCMF(int s,int t)
 {
     this->s=s;
     this->t=t;
     ent=;
     cost=;
     flow=;
     isq=new bool [MAXP];
     dist=new int[MAXP];
     head=new int[MAXP];
     edge=new Edge[MAXE];
     pre=new int[MAXE];
     memset(head,-,sizeof(int)*MAXP);
 }
 void MCMF::insert(int s,int t,int f,int c)
 {
     edge[ent].s=s;edge[ent].t=t;edge[ent].f=f;
     edge[ent].c=c;edge[ent].next=head[s];head[s]=ent++;
     edge[ent].s=t;edge[ent].t=s;edge[ent].f=;
     edge[ent].c=-c;edge[ent].next=head[t];head[t]=ent++;
 }
 bool MCMF::spfa()
 {
     memset(pre,-,sizeof(int)*MAXE);
     for(int i=;i<=t;i++)
     {
         isq[i]=false;
         dist[i]=INT_MAX;
     }
     queue<int>q;
     q.push(s);
     isq[s]=true;
     dist[s]=;
     while(!q.empty())
     {
         int temp1=q.front();
         q.pop();
         isq[temp1]=false;
         for(int i=head[temp1];i!=-;i=edge[i].next)
         {
             int temp2=edge[i].t;
             if(edge[i].f&&edge[i].c+dist[temp1]<dist[temp2])
             {
                 dist[temp2]=edge[i].c+dist[temp1];
                 pre[temp2]=i;
                 if(!isq[temp2])
                     q.push(temp2);
             }
         }
     }
     return pre[t]!=-;
 }
 void MCMF::mcmf()
 {
     while(spfa())
     {
         int minn=INT_MAX;
         for(int i=pre[t];i!=-;i=pre[i])
         {
             minn=MIN(minn,edge[i].f);
             i=edge[i].s;
         }
         flow+=minn;
         for(int i=pre[t];i!=-;i=pre[i])
         {
             edge[i].f-=minn;
             edge[i^].f+=minn;
             i=edge[i].s;
         }
         cost+=minn*dist[t];
     }
     cout<<-cost<<endl;
 }
 void MCMF::print()
 {
     for(int i=head[n+];i!=-;i=edge[i].next)
         if(edge[i].f==&&edge[i].t!=n+)
             printf("%d ",edge[i].t-);
     printf("\n");
     for(int i=head[n+];i!=-;i=edge[i].next)
         if(edge[i].f==&&edge[i].t!=n+)
             printf("%d ",edge[i].t-);
 }
 int a,b;
 int main()
 {
     int s,t,c;
     scanf("%d%d%d",&n,&a,&b);
     MCMF *temp=new MCMF(,n+);
     for(int i=;i<=n;i++)
         temp->insert(,i+,,);
     for(int i=;i<=n;i++)
     {
         scanf("%d",&m);
         temp->insert(i+,n+,,-m);
     }
     for(int i=;i<=n;i++)
     {
         scanf("%d",&m);
         temp->insert(i+,n+,,-m);
     }
     temp->insert(n+,n+,a,);
     temp->insert(n+,n+,b,);
     temp->mcmf();
     temp->print();
     return ;
 }