首页 > 代码库 > 【PowerOJ1740】 圆桌问题
【PowerOJ1740】 圆桌问题
https://www.oj.swust.edu.cn/problem/show/1740 (题目链接)
题意
n个单位的人去吃饭,m张餐桌,同一单位的人不能在同一餐桌,问可行方案。
Solution
最大流。
新建源点S,向每个单位连一条容量为R[i]的边;新建汇点T,从每张餐桌连一条容量为C[i]的边。单位和餐桌两两之间连一条容量为1的边。
代码
// PowerOJ1740#include<algorithm>#include<iostream>#include<cstring>#include<cstdlib>#include<cstdio>#include<cmath>#include<queue>#define LL long long#define inf 2147483640#define Pi acos(-1.0)#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);using namespace std;const int maxn=1000;struct edge {int to,next,w;}e[100010];int head[maxn],d[maxn],a[maxn],c[maxn];int n,m,cnt=1,sum,es,et,ans;void link(int u,int v,int w) { e[++cnt]=(edge){v,head[u],w};head[u]=cnt; e[++cnt]=(edge){u,head[v],0};head[v]=cnt;}bool bfs() { memset(d,-1,sizeof(d)); queue<int> q;q.push(es);d[es]=0; while (!q.empty()) { int x=q.front();q.pop(); for (int i=head[x];i;i=e[i].next) if (e[i].w && d[e[i].to]<0) { d[e[i].to]=d[x]+1; q.push(e[i].to); } } return d[et]>0;}int dfs(int x,int f) { if (x==et || f==0) return f; int w,used=0; for (int i=head[x];i;i=e[i].next) if (e[i].w && d[e[i].to]==d[x]+1) { w=dfs(e[i].to,min(e[i].w,f-used)); used+=w; e[i].w-=w;e[i^1].w+=w; if (used==f) return used; } if (!used) d[x]=-1; return used;}void Dinic() { while (bfs()) ans+=dfs(es,inf);}void print() { if (ans!=sum) {printf("0");return;} puts("1"); for (int i=1;i<=n;i++) { for (int j=head[i];j;j=e[j].next) if (e[j].w==0 && e[j].to<n+m+1) printf("%d ",e[j].to-n); puts(""); }}int main() { scanf("%d%d",&n,&m); for (int i=1;i<=n;i++) scanf("%d",&a[i]),sum+=a[i]; for (int i=1;i<=m;i++) scanf("%d",&c[i]); es=n+m+1;et=n+m+2; for (int i=1;i<=n;i++) { link(es,i,a[i]); for (int j=1;j<=m;j++) link(i,n+j,1); } for (int j=1;j<=m;j++) link(n+j,et,c[j]); Dinic(); print(); return 0;}
【PowerOJ1740】 圆桌问题
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。