标签:bfs names include rip vector 网络流24题 empty stdin esc
Description
假设有来自$n$个不同单位的代表参加一次国际会议。每个单位的代表数分别为$r_i(i\;\in\;[1,n])$。会议餐厅共有$m$张餐桌,每张餐桌可容纳$c_i(i\;\in\;[1,m])$个代表就餐。为了使代表们充分交流,希望从同一个单位来的代表不在同一个餐桌就餐。求满足要求的代表就餐方案。
Input
第$1$行有$2$个正整数$m,n.m$表示单位数,$n$表示餐桌数。
第$2$行有$m$个正整数,分别表示每个单位的代表数。
文件第$3$行有$n$个正整数,分别表示每个餐桌的容量。
Output
如果问题有解,第$1$行输出$1$,否则输出$0$。
接下来的$m$行给出每个单位代表的就餐桌号。如果有多个满足要求的方案,只要输出$1$个方案。
Sample Input
4 5
4 5 3 5
3 5 2 6 4
Sample Output
1
1 2 4 5
1 2 3 4 5
2 4 5
1 2 3 4 5
HINT
$1\;\leq\;m\;\leq\;150,1\;\leq\;n\;\leq\;270$
Solution
单位$i$为$x_i$,餐桌$i$为$y_i$,建立二分图.
从$s$向$x_i$连接一条容量为$r_i$的有向边,
从$x_i$向$y_j$连接一条容量为$1$的有向边,
从$y_i$向$t$连接一条容量为$c_i$的有向边,
如果最大流=人数和,则有解,从集合$x$从发流向集合$y$的满流边为当前方案.
#include<cmath> #include<ctime> #include<queue> #include<stack> #include<cstdio> #include<vector> #include<cstring> #include<cstdlib> #include<iostream> #include<algorithm> #define N 275 #define M 155 using namespace std; struct graph{ int nxt,to,f; }e[(N*M)<<1]; int a[M],b[N],g[N+M],dep[N+M],n,m,s,t,sum,cnt=1; queue<int> q; inline void addedge(int x,int y,int f){ e[++cnt].nxt=g[x];g[x]=cnt;e[cnt].to=y;e[cnt].f=f; } inline void adde(int x,int y,int f){ addedge(x,y,f);addedge(y,x,0); } inline bool bfs(int u){ memset(dep,0,sizeof(dep)); dep[u]=1;q.push(u); while(!q.empty()){ u=q.front();q.pop(); for(int i=g[u];i;i=e[i].nxt) if(e[i].f>0&&!dep[e[i].to]){ q.push(e[i].to); dep[e[i].to]=dep[u]+1; } } return dep[t]; } inline int dfs(int u,int f){ int ret=0; if(u==t) return f; for(int i=g[u],d;i&&f;i=e[i].nxt) if(e[i].f>0&&dep[e[i].to]>dep[u]){ d=dfs(e[i].to,min(f,e[i].f)); e[i].f-=d;e[i^1].f+=d;ret+=d;f-=d; } return ret; } inline int dinic(){ int ret=0; while(true){ if(!bfs(s)) return ret; ret+=dfs(s,sum); } } inline void Aireen(){ scanf("%d%d",&m,&n); s=n+m+1;t=s+1; for(int i=1;i<=m;++i){ scanf("%d",&a[i]); adde(s,i,a[i]); sum+=a[i]; } for(int i=1;i<=n;++i){ scanf("%d",&b[i]); adde(i+m,t,b[i]); } for(int i=1;i<=m;++i) for(int j=n;j;--j) adde(i,j+m,1); if(dinic()==sum){ puts("1"); for(int i=1;i<=m;++i){ for(int j=g[i];j;j=e[j].nxt) if(!(j&1)&&!e[j].f) printf("%d ",e[j].to-m); printf("\n"); } } else puts("0"); } int main(){ freopen("table.in","r",stdin); freopen("table.out","w",stdout); Aireen(); fclose(stdin); fclose(stdout); return 0; }
标签:bfs names include rip vector 网络流24题 empty stdin esc
原文地址:http://www.cnblogs.com/AireenYe/p/6240832.html