网络流24题之圆桌问题

对于每个单位连源点到其的边，流量为单位人数

每个单位对每个桌子连边，流量为1

每个桌子对终点连边，流量为桌子人数

判断最大匹配是不是为总人数

By：大奕哥

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int N=20005;
 4 int head[N],d[N],a[N],c[N];
 5 int n,m,cnt=-1,s,t,sum;
 6 struct node{
 7     int to,nex,w;
 8 }e[1000005];
 9 void add(int x,int y,int w)
10 {
11     e[++cnt].to=y;e[cnt].nex=head[x];head[x]=cnt;e[cnt].w=w;
12     e[++cnt].to=x;e[cnt].nex=head[y];head[y]=cnt;e[cnt].w=0;
13 }
14 queue<int>q;
15 bool bfs(int x,int y)
16 {
17     memset(d,-1,sizeof(d));
18     d[x]=0;q.push(x);
19     while(!q.empty())
20     {
21         int x=q.front();q.pop();
22         for(int i=head[x];i!=-1;i=e[i].nex)
23         {
24             int y=e[i].to;
25             if(d[y]!=-1||!e[i].w)continue;
26             d[y]=d[x]+1;q.push(y);
27         }
28     }
29     return d[y]!=-1;
30 }
31 int dfs(int x,int w,int yy)
32 {
33     if(x==yy||!w)return w;
34     int s=0;
35     for(int i=head[x];i!=-1;i=e[i].nex)
36     {
37         int y=e[i].to;
38         if(!e[i].w||d[y]!=d[x]+1)continue;
39         int flow=dfs(y,min(w-s,e[i].w),yy);
40         if(!flow){
41             d[y]=-1;continue;
42         }
43         e[i].w-=flow;e[i^1].w+=flow;s+=flow;
44         if(s==w)return s;
45     }
46     return s;
47 }
48 int dinic()
49 {
50     int ans=0;
51     while(bfs(s,t))
52     {
53         ans+=dfs(s,1e9,t);
54     }
55     return ans;
56 }
57 int main()
58 {
59     scanf("%d%d",&m,&n);
60     s=0;t=n+m+10;memset(head,-1,sizeof(head));
61     for(int i=1;i<=m;++i)
62     {
63         scanf("%d",&a[i]);
64         add(s,i,a[i]);sum+=a[i];
65         for(int j=1;j<=n;++j)
66         add(i,j+m,1);
67     }
68     for(int i=1;i<=n;++i)
69     {
70         scanf("%d",&c[i]);
71         add(i+m,t,c[i]);
72     }
73     int ans=dinic();
74     if(ans!=sum)
75     {
76         printf("0");return 0;
77     }
78     puts("1");
79     for(int i=1;i<=m;++i)
80     {
81         for(int j=head[i];j!=-1;j=e[j].nex)
82         {
83             if(e[j].w)continue;
84             printf("%d ",e[j].to-m);
85         }
86         printf("\n");
87     }
88     return 0;
89 }

 

posted @ 2018-01-31 17:49  大奕哥&VANE  阅读(147)  评论(0)  编辑  收藏  举报