洛谷 P1402 酒店之王(dinic模板)

将1个人拆成2个点,常规建图。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int maxn=450;
const int INF=0x3f3f3f3f;
int n,p,q,S=0,T;

int head[maxn],Ecnt;
struct edge{
    int to,nxt,w;
}G[maxn*maxn*2];

inline void addE(int u,int v,int w){
    G[Ecnt]=(edge){v,head[u],w};
    head[u]=Ecnt++;
    G[Ecnt]=(edge){u,head[v],0};
    head[v]=Ecnt++;
}

int dep[maxn];
bool bfs(){
    queue <int> Q;
    memset(dep,-1,sizeof(dep));
    Q.push(S),dep[S]=1;
    while(!Q.empty()){
        int u=Q.front(); Q.pop();
        if(u==T) return 1;
        for(int i=head[u];i!=-1;i=G[i].nxt){
            int v=G[i].to;
            if(dep[v]!=-1||G[i].w==0) continue;
            Q.push(v),dep[v]=dep[u]+1;
        }
    }
    return 0;
}

int dfs(int u,int minw){
    int flow=0;
    if(u==T) return minw;
    for(int i=head[u];i!=-1;i=G[i].nxt){
        int v=G[i].to;
        if(dep[v]!=dep[u]+1||G[i].w==0) continue;
        int tmp=dfs(v,min(G[i].w,minw));
        flow+=tmp,minw-=tmp,G[i].w-=tmp,G[i^1].w+=tmp;
        if(minw==0) break;
    }
    return flow;
}

int dinic(){
    int ret=0;
    while(bfs()) ret+=dfs(S,INF);
    return ret;
}

int main(){
    memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&p,&q);
    T=p+q+n*2+1;
    for(int i=1;i<=p;i++)
        addE(S,i,1);
    for(int i=p+1;i<=p+q;i++)
        addE(i,T,1);
    for(int i=p+q+1;i<=p+q+n;i++)
        addE(i,i+n,1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=p;j++){
            int tmp;
            scanf("%d",&tmp);
            if(tmp) addE(j,p+q+i,1);
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=q;j++){
            int tmp;
            scanf("%d",&tmp);
            if(tmp) addE(p+q+n+i,p+j,1);
        }
    printf("%d\n",dinic());
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值