GYM 100694F - The Berland Championship （网络流）

题意

教练想要组个最强的队(￣▽￣)／(￣▽￣)／

现在有N个学生，M道题目，每个学生都有最多完成题目数量的限制P，以及他们所能解出的题目列表，选出3个学生，使他们能完成的题目最多。

解题思路

套个网络流模板写一下……

代码

#include<bits/stdc++.h>
#define N 55
using namespace std;
const int mod = (int)1e9+7;

int n,m,p[N],vis[N][1005],cnt[N];
char pro[1005],name[N][15];

struct Maxflow{
	static const int oo=100000005;
	static const int maxn=1111;//点数
	int T,S;
	struct node{
		int to,nxt,cap;
	}ed[maxn*2];
	int head[maxn],cur[maxn],cnt,ndnum;
	int h[maxn];
	void addedge(int u,int v,int cap){
		ed[cnt].to=v,ed[cnt].cap=cap,ed[cnt].nxt=head[u],head[u]=cnt++;
		ed[cnt].to=u,ed[cnt].cap=0,ed[cnt].nxt=head[v],head[v]=cnt++;
	}
	bool bfs(){
		memset(h,-1,sizeof(h));
		queue<int>q;
		h[S]=0,q.push(S);
		while(!q.empty()){
			int now=q.front();q.pop();
			for(int i=head[now];~i;i=ed[i].nxt){
				if(ed[i].cap&&h[ed[i].to]==-1){
					h[ed[i].to]=h[now]+1;
					q.push(ed[i].to);
				}
			}
		}
		return h[T]!=-1;
	}
	int dfs(int x,int f){
		if(x==T||f==0)return f;
		int w,used=0;
		for(int i=cur[x];~i;i=ed[i].nxt){
			if(h[ed[i].to]==h[x]+1){
				w=f-used;
				w=dfs(ed[i].to,min(w,ed[i].cap));
				ed[i].cap-=w;
				ed[i^1].cap+=w;
				if(ed[i].cap)cur[x]=i;
				used+=w;
				if(used==f)return f;
			}
		}
		if(!used)h[x]=-1;
		return used;
	}
	int dinic(){
	    int ans=0;
		while(bfs()){
			for(int i=1;i<=ndnum;i++){
				cur[i]=head[i];
			}
			ans+=dfs(S,oo);
		}
		return ans;
	}
	void init(int s,int t,int num){//起点，终点，点数 ，点从1开始
		for(int i=1;i<=num;i++)head[i]=-1;
		cnt=0;
		T=t;
		S=s;
		ndnum=num;
	}
}flow;


int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&p[i]);
    for(int i=1;i<=n;i++)scanf("%s",name[i]);
    for(int i=1;i<=n;i++){
        scanf("%s",pro+1);
        for(int j=1;j<=m;j++){
            if(pro[j]=='1')vis[i][j]=1;
        }
    }
    int ans=-1;
    int a=0,b=0,c=0;
    for(int i=1;i<=n;i++){
        for(int j=i+1;j<=n;j++){
            for(int k=j+1;k<=n;k++){
                memset(cnt,0,sizeof(cnt));
                flow.init(n+9,n+10,n+10);
                flow.addedge(n+9,i,p[i]);
                flow.addedge(n+9,j,p[j]);
                flow.addedge(n+9,k,p[k]);
                for(int l=1;l<=m;l++){
                    cnt[vis[i][l]+(vis[j][l]<<1)+(vis[k][l]<<2)]++;
                }
                for(int l=0;l<=7;l++){
                    flow.addedge(n+l+1,n+10,cnt[l]);
                    if(l&1)flow.addedge(i,n+l+1,mod);
					if(l&2)flow.addedge(j,n+l+1,mod);
					if(l&4)flow.addedge(k,n+l+1,mod);

                }
                int maxflow=flow.dinic();
                if(maxflow>ans){
                    ans=maxflow;
                    a=i;
                    b=j;
                    c=k;
                }
            }
        }
    }
    printf("%d\n",ans);
    printf("%s %s %s\n",name[a],name[b],name[c]);
    return 0;
}