2018.08.19 洛谷P1402 酒店之王(最大流)

时间:2024-08-22 20:05:32

传送门

最大流入门题,把人拆点即可。

代码:

#include<bits/stdc++.h>
#define N 505
using namespace std;
inline int read(){
    int ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
    return ans;
}
struct Node{int v,next,c;}e[N*N<<1];
int t,s,first[N],cnt=-1,d[N],n,p,q;
inline void add(int u,int v,int c){e[++cnt].v=v,e[cnt].c=c,e[cnt].next=first[u],first[u]=cnt;}
inline bool bfs(){
    queue<int>q;
    q.push(s);
    memset(d,-1,sizeof(d));
    d[s]=0;
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(int i=first[x];i!=-1;i=e[i].next){
            int v=e[i].v;
            if(e[i].c<=0||d[v]!=-1)continue;
            d[v]=d[x]+1;
            if(v==t)return true;
            q.push(v);
        }
    }
    return false;
}
inline int dfs(int x,int f){
    if(!f||x==t)return f;
    int flow=f;
    for(int i=first[x];i!=-1;i=e[i].next){
        int v=e[i].v;
        if(flow&&e[i].c>0&&d[v]==d[x]+1){
            int tmp=dfs(v,min(flow,e[i].c));
            if(tmp==0)d[v]=-1;
            flow-=tmp;
            e[i].c-=tmp;
            e[i^1].c+=tmp;
        }
    }
    return f-flow;
}
int main(){
    memset(first,-1,sizeof(first));
    n=read(),p=read(),q=read(),s=0,t=p+q+n*2+1;
    for(int i=1;i<=p;++i)add(s,i,1),add(i,s,0);
    for(int i=1;i<=q;++i)add(i+p,t,1),add(t,i+p,0);
    for(int i=1;i<=n;++i)add(i+p+q,i+p+q+n,1),add(i+p+q+n,i+p+q,0);
    for(int i=1;i<=n;++i)
        for(int j=1;j<=p;++j){
            int x=read();
            if(x)add(j,i+p+q,1),add(i+p+q,j,0);
        }
    for(int i=1;i<=n;++i)
        for(int j=1;j<=q;++j){
            int x=read();
            if(x)add(i+p+q+n,j+p,1),add(j+p,i+p+q+n,0);
        }
    int ans=0;
    while(bfs())ans+=dfs(s,0x3f3f3f3f);
    cout<<ans;
    return 0;
}