【二分圖/網路流】Gym - 101873F - Plug It In
阿新 • • 發佈:2018-11-10
題目連結<http://codeforces.com/gym/229936/problem/F>
題意:
有m個插座,n個電器,每個插座最多可連線一個電器。另外有一個插頭,可以使得一個插座連線三個電器,問最大匹配數是多少。
題解:
二分圖:二分圖做法相對簡潔一些,也會更快,儲存原先的最大匹配,列舉每個插座是否還存在增廣路。每次就是要對原圖的反覆複製。
#include<cstdio> #include<cstring> #include<queue> #include<cmath> #include<iostream> using namespace std; const int N=1e5+7; const int inf=1<<26; struct Edge{ int u,v,nxt; Edge(int u=0,int v=0,int nxt=0):u(u),v(v),nxt(nxt){} }edge[30*N]; int n,m,k,edn; int p[N],d[N],c[N]; void add(int u,int v){ edge[++edn]=Edge(u,v,p[u]);p[u]=edn; } int match[N],vis[N],tmp[N]; bool dfs(int u){ for(int i=p[u];~i;i=edge[i].nxt){ int v=edge[i].v; if(vis[v]) continue;vis[v]=1; if(match[v]==-1||dfs(match[v])){ match[v]=u; return true; } } return false; } int main() { memset(p,-1,sizeof(p));edn=-1; scanf("%d%d%d",&n,&m,&k); int u,v; for(int i=1;i<=k;i++){ scanf("%d%d",&u,&v); add(u,v); } int maxn=0; memset(match,-1,sizeof(match)); for(int i=1;i<=n;i++){ memset(vis,0,sizeof(vis)); if(dfs(i)) maxn++; } memcpy(tmp,match,sizeof(match)); int ans=0; for(int i=1;i<=n;i++){ int cnt=0; for(int j=1;j<=2;j++){ memset(vis,0,sizeof(vis)); if(dfs(i)) cnt++; else break; } ans=max(ans,cnt); memcpy(match,tmp,sizeof(tmp)); } printf("%d\n",ans+maxn); return 0; }
網路流:網路流其實沒那麼必要,各種複雜度都很大。
先跑一遍容量都為1的dinic,然後列舉插座多給兩個容量,在殘餘網路中跑。
麻煩的是網路流如果直接複製原圖,那麼加上本身的複雜度會TLE。但也可以發現每次我只增加了2個容量,相對於原來的殘留網路變化其實不大,只要把這些變化的流量記錄在一個數組裡,逐一退回即可。
#include<cstdio> #include<cstring> #include<queue> #include<cmath> #include<iostream> using namespace std; const int N=1e5+7; const int inf=1<<26; struct Edge{ int u,v,w,nxt; Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){} }edge[30*N]; int n,m,k,edn,sp,tp; int p[N],d[N],c[N]; void add(int u,int v,int w){ edge[++edn]=Edge(u,v,w,p[u]);p[u]=edn; edge[++edn]=Edge(v,u,0,p[v]);p[v]=edn; } bool bfs(){ memset(d,-1,sizeof(d));d[sp]=0; queue<int>q;q.push(sp); while(!q.empty()){ int u=q.front(); q.pop(); for(int i=p[u];~i;i=edge[i].nxt){ int v=edge[i].v; if(d[v]==-1&&edge[i].w){ d[v]=d[u]+1; q.push(v); if(v==tp) return true; } } } return ~d[tp]; } struct chg{ int id,val; }cg[N*2]; int top; int dfs(int u,int b){ if(u==tp) return b; int r=0; for(int i=c[u];~i;i=edge[i].nxt){ int v=edge[i].v; if(edge[i].w&&d[v]==d[u]+1){ int x=min(edge[i].w,b-r); c[u]=i; x=dfs(v,x); r+=x; edge[i].w-=x; edge[i^1].w+=x; cg[++top].id=i; cg[top].val=x; if(r==b) break; } } if(!r)d[u]=-2; return r; } int dinic(){ int total=0,t; while(bfs()){ memcpy(c,p,sizeof(p)); while(t=dfs(sp,inf)) total+=t; } return total; } int main() { memset(p,-1,sizeof(p));edn=-1; scanf("%d%d%d",&n,&m,&k); sp=n+m+1,tp=sp+1; int u,v; for(int i=1;i<=k;i++){ scanf("%d%d",&u,&v); add(u,v+n,1); } for(int i=1;i<=n;i++) add(sp,i,1); for(int i=1;i<=m;i++) add(i+n,tp,1); int ans=dinic(),tmp=0; for(int j=1;j<=n;j++){ top=0; add(sp,j,2); tmp=max(tmp,dinic()); for(int i=1;i<=top;i++){ int id=cg[i].id; int val=cg[i].val; edge[id].w+=val; edge[id^1].w-=val; } p[sp]=edge[p[sp]].nxt; p[j]=edge[p[j]].nxt; edn-=2; } printf("%d\n",ans+tmp); return 0; }