很经典的模型了,之前博客讲的很详细了。
现在说下这题wa5的坑:
输出方案时容量为0的边可能是从起点连过来的边的反向边!!
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M = 1000000+7;
const int N = 1100+7;
struct Dinic{
//N个点,M条边
//add建双向边,然后D.gao ,最后输出maxflow
#define inf 0x3f3f3f3f
ll maxflow;int s,t,n;
int v[N],pre[N],d[N],now[N];
ll incf[N];
int head[N],cnt=1;
struct EDGE{int to,nxt;ll w;}ee[M*2];
inline void AD(int x,int y,ll w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}
inline void add(int x,int y,ll w){AD(x,y,w);AD(y,x,0);}
inline bool bfs()//在残量网络上构造分层图
{
memset(d,0,sizeof(d));
queue<int>q;
q.push(s);d[s]=1;
while(q.size())
{
int x=q.front();q.pop();
for(int i=head[x];i;i=ee[i].nxt)
{
int y=ee[i].to;ll w=ee[i].w;
if(w&&!d[y])
{
q.push(y);
d[y]=d[x]+1;
if(y==t)return 1;
}
}
}
return false;
}
inline int dinic(int x,int flow)
{
if(x==t)return flow;
ll rest = flow,k;
for(int i=now[x];i&&rest;i=ee[i].nxt)
{
int y=ee[i].to;ll w= ee[i].w;
now[x]=i;
if(w&&d[y]==d[x]+1)
{
k=dinic(y,min(rest,w));
if(!k)d[y]=0;//剪枝,去掉增广完毕的点
ee[i].w-=k;
ee[i^1].w+=k;
rest-=k;
}
}
return flow - rest;
}
inline void gao()
{
int flow=0;
while(bfs())
{
for(int i=0;i<=n;i++)now[i]=head[i];
while(flow=dinic(s,inf))maxflow+=flow;
}
}
inline void init(int nn,int S,int T)
{
cnt=1;maxflow=0;
for(int i=0;i<=n;i++)head[i]=0;
s=S,t=T,n=nn;
}
}D;
//求解二分图匹配问题时的复杂度是:m*sqrt(n)的
int main()
{
int n,m,s,t,x,l;
cin>>n>>m;
int N =n+m+2,sm=0;
s=n+m+1,t=n+m+2;
D.init(N,s,t);//初始化节点个数,起点,终点
for(int i=1;i<=n;i++){
cin>>x;sm+=x;
D.add(s,i,x);
}
for(int i=1;i<=m;i++)
{
cin>>l;
if(l)D.add(i+n,t,1);
for(int j=1;j<=l;j++){
cin>>x;
D.add(x,i+n,1);
}
}
D.gao();
if(D.maxflow!=sm){
cout<<"No Solution!"<<endl;
return 0;
}
for(int x=1;x<=n;x++){
cout<<x<<":";
for(int j=D.head[x];j;j=D.ee[j].nxt){
int y = D.ee[j].to,w=D.ee[j].w;
if(!w&&y!=s)cout<<" "<<y-n;//可能是流向起点的反向边
}
cout<<endl;
}
return 0;
}