【洛谷P1402】酒店之王
阿新 • • 發佈:2019-04-26
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】酒店之王