4 3 3 1 1 1 1 1 1 YYN NYY YNY YNY YNY YYN YYN NNY
3
题意:有N头牛,F种食物,D种饮料;
有2*N组输入,第一组1-N表示第i头牛对F种食物的喜欢与否,Y表示喜欢没N表示不喜欢,第二组1-N表示对饮料的喜欢与否;
问最多有几头牛可以吃上自己喜欢的食物,并喝上自己喜欢的饮料;
最大流算法, S——F(food)——N(n头牛)——N(n头牛)——D(饮料)——T;S表示超级源点,T表示超级汇点;
#include <iostream> #include <queue> #include <stdio.h> #include <string.h> #include <math.h> #include <vector> #include <stdlib.h> const int inf=0x3f3f3f3f; const int MAXN=1e5+10; const int MAXM=400010; using namespace std; struct Node { int to,next,cap; }edge[MAXM]; int tol;int head[MAXN]; int gap[MAXN],dis[MAXN],pre[MAXN],cur[MAXN]; void init() { tol=0; memset(head,-1,sizeof(head)); } void addedge(int u,int v,int w,int rw=0) { edge[tol].to=v;edge[tol].cap=w;edge[tol].next=head[u];head[u]=tol++; edge[tol].to=u;edge[tol].cap=rw;edge[tol].next=head[v];head[v]=tol++; } int sap(int start,int end0,int nodenum) { memset(dis,0,sizeof(dis)); memset(gap,0,sizeof(gap)); memcpy(cur,head,sizeof(head)); int u=pre[start]=start,maxflow=0,aug=-1; gap[0]=nodenum; while(dis[start]<nodenum) { loop: for(int &i=cur[u];i!=-1;i=edge[i].next) { int v=edge[i].to; if(edge[i].cap&&dis[u]==dis[v]+1) { if(aug==-1||aug>edge[i].cap) aug=edge[i].cap; pre[v]=u; u=v; if(v==end0) { maxflow+=aug; for(u=pre[u];v!=start;v=u,u=pre[u]) { edge[cur[u]].cap-=aug; edge[cur[u]^1].cap+=aug; } aug=-1; } goto loop; } } int mindis=nodenum; for(int i=head[u];i!=-1;i=edge[i].next) { int v=edge[i].to; if(edge[i].cap&&mindis>dis[v]) { cur[u]=i; mindis=dis[v]; } } if((--gap[dis[u]])==0)break; gap[dis[u]=mindis+1]++; u=pre[u]; } return maxflow; } int S,T; char str[555]; int N,F,D; int food[MAXN],dirk[MAXN]; int main() { while(scanf("%d%d%d",&N,&F,&D)!=-1) { S=0; T=F+D+N*2+1; init(); for(int i=1; i<=F; i++) { scanf("%d",&food[i]); // cin>>food[i]; addedge(S,i,food[i]); } for(int i=1; i<=D; i++) { scanf("%d",&dirk[i]); // cin>>dirk[i]; addedge(i+F+N*2,T,dirk[i]); } for(int i=1; i<=N; i++) addedge(i+F,i+N+F,1); for(int i=1; i<=N; i++) { scanf("%s",str); //cin>>str; int l=strlen(str); for(int j=0; j<l; j++) { if(str[j]=='Y') addedge(j+1,F+i,1); } } for(int i=1; i<=N; i++) { scanf("%s",str); //cin>>str; int l=strlen(str); for(int j=0; j<l; j++) { if(str[j]=='Y') addedge(F+N+i,F+N*2+j+1,1); } } printf("%d\n",sap(S,T,T+1)); //cout<<max_flow(0,hui)<<endl; } return 0; }