1. 程式人生 > >【洛谷P1402】酒店之王

【洛谷P1402】酒店之王

head res std 連接 nic read %d using i++

題目大意:有三個集合 \(P,Q,N\),P 與 N 集合之間存在一些有向邊,N 與 Q 集合之間存在一些有向邊。在三個集合中每個點最多只能利用一次的前提下,求最多能利用多少N 集合中的點,使得 \((p,n,q)\) 三個節點之間均有邊相連。

題解:若只有兩個集合,顯然是二分圖的最大匹配問題。對於三個集合來說,最麻煩的問題在於 N 集合中每個點只能被選一次,但是一個 N 集合中的節點卻可能對應著多個P、Q 集合中的節點。保證 N 只能被選一次的方法是:將 N 集合中的點拆分成兩個點,分別表示入點和出點,且連接入點和出點之間的邊權為 1。

代碼如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=410;
const int maxm=1e5+10;

int n,p,q,s,t,maxflow,d[maxn];
struct node{int nxt,to,w;}e[maxm<<1];
int tot=1,head[maxn];
inline void add_edge(int from,int to,int w){
    e[++tot]=node{head[from],to,w},head[from]=tot;
    e[++tot]=node{head[to],from,0},head[to]=tot;
}

bool bfs(){
    queue<int> q;
    memset(d,0,sizeof(d));
    d[s]=1,q.push(s);
    while(q.size()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=e[i].nxt){
            int v=e[i].to,w=e[i].w;
            if(w&&!d[v]){
                d[v]=d[u]+1;
                q.push(v);
                if(v==t)return 1;
            }
        }
    }
    return 0;
}
int dinic(int u,int flow){
    if(u==t)return flow;
    int rest=flow;
    for(int i=head[u];i&&rest;i=e[i].nxt){
        int v=e[i].to,w=e[i].w;
        if(w&&d[v]==d[u]+1){
            int k=dinic(v,min(flow,w));
            if(!k)d[v]=0;
            e[i].w-=k,e[i^1].w+=k,rest-=k;
        }
    }
    return flow-rest;
}

void read_and_parse(){
    scanf("%d%d%d",&n,&p,&q);
    s=0,t=2*n+p+q+1;
    for(int i=1;i<=n;i++)add_edge(i,i+n,1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=p;j++){
            int x;scanf("%d",&x);
            if(x)add_edge(j+2*n,i,1);
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=q;j++){
            int x;scanf("%d",&x);
            if(x)add_edge(i+n,2*n+p+j,1);
        }
    for(int i=2*n+1;i<=2*n+p;i++)add_edge(s,i,1);
    for(int i=2*n+p+1;i<=2*n+p+q;i++)add_edge(i,t,1);
}
void solve(){
    int now;
    while(bfs())while(now=dinic(s,0x3f3f3f3f))maxflow+=now;
    printf("%d\n",maxflow);
}
int main(){
    read_and_parse();
    solve();
    return 0;
}

【洛谷P1402】酒店之王