首页 > 代码库 > 专题之匹配、网络流(一)

专题之匹配、网络流(一)

1、hdu 2444 The Accomodation of Students(判断二分图+最大匹配)(匈牙利模板)

  题意:一共有n个学生,m对关系:A认识B。问能否将所有的人分成两批,每批之间的人都互相认识,如果可以,输出每批的人数。即判断是否为二分图,以及求二分图的最大匹配。

  思路:判断是否为二分图(DFS或BFS);求二分图的最大匹配:匈牙利算法。

技术分享
 1 #include<iostream>
 2 #include<queue>
 3 using namespace std;
 4 int n,m;
 5 const int maxn = 210;//x集合和y集合总最大的点数
 6 bool mp[maxn][maxn];//1表示该ij可以匹配
 7 int cx[maxn];//记录x集合中匹配的y元素是哪一个
 8 int cy[maxn];//记录y集合中匹配的x元素是哪一个
 9 int vis[maxn];//标记该顶点是否访问过
10 int cntx;
11 bool dfs(int u)
12 {
13     for (int v = 1; v <= n; v++)//两个集合内共有n个元素
14     {
15         if (mp[u][v] && !vis[v])
16         {
17             vis[v] = 1;
18             if (cy[v] == -1 || dfs(cy[v]))//)//如果y集合中的v元素没有匹配或者是v已经匹配,但是从cy[v]中能够找到一条增广路
19             {
20                 cx[u] = v; cy[v] = u;
21                 return 1;
22             }
23         }
24     }
25     return 0;
26 }
27 int maxmatch()//匈牙利算法主函数
28 {
29     int ans = 0;
30     memset(cx, 0xff, sizeof cx);//初始值为-1表示两个集合中都没有匹配的元素!
31     memset(cy, 0xff, sizeof cy);
32     for (int i = 1; i <= n; i++)
33         if (cx[i] == -1)//如果i未匹配
34         {
35             memset(vis, 0, sizeof(vis));
36             ans += dfs(i);
37         }
38     return ans/2;//对两个部里的都匹配了,这样就相当于匹配了两次了  
39 }
40 bool istwo()
41 {//判断是否为二分图
42     queue<int>q;
43     memset(vis, 0, sizeof(vis));
44     q.push(1);
45     vis[1] = true;
46     while (!q.empty())
47     {
48         int u = q.front();
49         q.pop();
50         for (int i = 1; i <= n; i++)
51         {
52             if (mp[u][i])
53             {
54                 if (vis[i] == 0)
55                 {
56                     if (vis[u] == 1) vis[i] = 2;
57                     else vis[i] = 1;
58                     q.push(i);
59                 }
60                 else
61                 {
62                     if (vis[i] == vis[u]) return false;
63                 }
64             }
65         }
66     }
67     return true;
68 }
69 int main()
70 {
71     while (~scanf("%d%d", &n, &m))
72     {
73         memset(mp ,0, sizeof(mp));
74         while (m--)
75         {
76             int a, b;
77             scanf("%d%d", &a, &b);
78             mp[a][b] = mp[b][a] = 1;
79         }
80         if (!istwo()|| n == 1)
81         {
82             printf("No\n");
83         }
84         else
85         {
86             int ans = maxmatch();
87             printf("%d\n", ans);
88         }
89     }
90 
91     return 0;
92 }
View Code

2、hdu 1083 Courses(最大匹配)

  题意:有P种课程,N个学生。接下来P行,第i行第一个Ni表示喜欢第i个课程的学生的人数,接下来是Ni个学生。问:能否有一种匹配使得每个学生都选一门不同的课程,同时所有课程都出现。

  思路:二分图最大匹配,匈牙利算法,课程为x集,学生为y集。判断最大匹配数是否为P。

技术分享
 1 #include<iostream>
 2 #include<queue>
 3 #include<memory.h>
 4 using namespace std;
 5 int n, p;
 6 const int maxn = 310;//最大学生数
 7 const int maxp = 110;//最大学科数
 8 bool mp[maxp][maxn];//1表示该ij可以匹配
 9 int cx[maxp];//记录x集合中匹配的y元素是哪一个
10 int cy[maxn];//记录y集合中匹配的x元素是哪一个
11 int vis[maxn];//标记该顶点是否访问过
12 bool dfs(int u)
13 {
14     for (int v = 1; v <= n; v++)
15     {
16         if (mp[u][v] && !vis[v])
17         {
18             vis[v] = 1;
19             if (cy[v] == -1 || dfs(cy[v]))//)//如果y集合中的v元素没有匹配或者是v已经匹配,但是从cy[v]中能够找到一条增广路
20             {
21                 cx[u] = v; cy[v] = u;
22                 return 1;
23             }
24         }
25     }
26     return 0;
27 }
28 int maxmatch()//匈牙利算法主函数
29 {
30     int ans = 0;
31     memset(cx, 0xff, sizeof cx);//初始值为-1表示两个集合中都没有匹配的元素!
32     memset(cy, 0xff, sizeof cy);
33     for (int i = 1; i <= p; i++)
34         if (cx[i] == -1)//如果i未匹配
35         {
36             memset(vis, 0, sizeof(vis));
37             ans += dfs(i);
38         }
39     return ans;
40 }
41 
42 int main()
43 {
44     int t;
45     scanf("%d", &t);
46     while (t--)
47     {
48         scanf("%d%d", &p, &n);
49         memset(mp, 0, sizeof(mp));
50         for (int i = 1; i <= p; i++)
51         {
52             int count;
53             scanf("%d", &count);
54             for (int j = 1; j <= count; j++)
55             {
56                 int v;
57                 scanf("%d", &v);
58                 mp[i][v] = true;
59             }
60        }
61         int ans = maxmatch();
62         if (ans < p)printf("NO\n");
63         else printf("YES\n");
64     }
65 
66     return 0;
67 }
View Code

3、hdu 1281 棋盘游戏(最大匹配)

  题意:所有棋子不能同行或同列。给出棋子可以放的位置,问最多能放几个,同时给出重要点(去掉某个位置后就无法保证放尽量多的棋子)的个数。

  思路:x集代表行,y集代表列,对于可以放的位置(x,y)将x和y连一条边。最多能放的棋子即该二分图的最大匹配。接下来每次去掉一个位置看看最大匹配是否不变,如果改变则为重要点。

技术分享
 1 #include<iostream>
 2 #include<queue>
 3 #include<memory.h>
 4 using namespace std;
 5 int n, m, k;
 6 const int maxr = 110;//最大行数
 7 const int maxc = 110;//最大列数
 8 const int maxk = 10010;//最多可放位置
 9 bool mp[maxr][maxc];//1表示该ij可以匹配
10 int cx[maxc];//记录x集合中匹配的y元素是哪一个
11 int cy[maxr];//记录y集合中匹配的x元素是哪一个
12 int vis[maxr];//标记该顶点是否访问过
13 struct point
14 {
15     int x;
16     int y;
17 }points[maxk];
18 bool dfs(int u)
19 {
20     for (int v = 1; v <= m; v++)
21     {
22         if (mp[u][v] && !vis[v])
23         {
24             vis[v] = 1;
25             if (cy[v] == -1 || dfs(cy[v]))//)//如果y集合中的v元素没有匹配或者是v已经匹配,但是从cy[v]中能够找到一条增广路
26             {
27                 cx[u] = v; cy[v] = u;
28                 return 1;
29             }
30         }
31     }
32     return 0;
33 }
34 int maxmatch()//匈牙利算法主函数
35 {
36     int ans = 0;
37     memset(cx, 0xff, sizeof cx);//初始值为-1表示两个集合中都没有匹配的元素!
38     memset(cy, 0xff, sizeof cy);
39     for (int i = 1; i <= n; i++)
40         if (cx[i] == -1)//如果i未匹配
41         {
42             memset(vis, 0, sizeof(vis));
43             ans += dfs(i);
44         }
45     return ans;
46 }
47 
48 int main()
49 {
50     int Case = 1;
51     while (~scanf("%d%d%d", &n, &m, &k))
52     {
53         memset(mp, 0, sizeof(mp));
54         for (int i = 1; i <= k; i++)
55         {
56             scanf("%d%d", &points[i].x, &points[i].y);
57             mp[points[i].x][points[i].y] = 1;
58         }
59         int ans = maxmatch();
60         int import = 0;
61         for (int i = 1; i <= k; i++)
62         {
63             mp[points[i].x][points[i].y] = 0;
64             int tmp = maxmatch();
65             mp[points[i].x][points[i].y] = 1;
66             if (tmp < ans) import++;
67         }
68         printf("Board %d have %d important blanks for %d chessmen.\n", Case++, import, ans);
69     }
70 
71     return 0;
72 }
View Code

4、HDU 2819 Swap(最大匹配)

  题意:给出n*n的矩阵,问能否通过交换某两行或某两列若干次后,使得其对角线上元素均为1.

  思路:只交换行或只交换列均可达到一样的效果(如果达不到,说明无解)。列既作为x集,也作为y集,[i][j]连线表示将第j列和第i列互换(当[x][y]为1时,将[x][y]连线,表示将可以其换到[x][x])。然后求此二分图的最大匹配。输出每个匹配时,注意交换,否则会多输出。匈牙利算法。

技术分享
 1 #include<iostream>
 2 #include<queue>
 3 #include<memory.h>
 4 #include<algorithm>
 5 using namespace std;
 6 int n;
 7 const int maxr = 110;//最大行数
 8 const int maxc = 110;//最大列数
 9 const int maxk = 1010;//最多可放位置
10 bool mp[maxr][maxc];//1表示该ij可以匹配
11 int cx[maxc];//记录x集合中匹配的y元素是哪一个
12 int cy[maxr];//记录y集合中匹配的x元素是哪一个
13 int vis[maxr];//标记该顶点是否访问过
14 struct stp
15 {
16     int x;
17     int y;
18 }stps[maxk];
19 bool dfs(int u)
20 {
21     for (int v = 1; v <= n; v++)
22     {
23         if (mp[u][v] && !vis[v])
24         {
25             vis[v] = 1;
26             if (cy[v] == -1 || dfs(cy[v]))//)//如果y集合中的v元素没有匹配或者是v已经匹配,但是从cy[v]中能够找到一条增广路
27             {
28                 cx[u] = v; cy[v] = u;
29                 return 1;
30             }
31         }
32     }
33     return 0;
34 }
35 int maxmatch()//匈牙利算法主函数
36 {
37     int ans = 0;
38     memset(cx, 0xff, sizeof cx);//初始值为-1表示两个集合中都没有匹配的元素!
39     memset(cy, 0xff, sizeof cy);
40     for (int i = 1; i <= n; i++)
41         if (cx[i] == -1)//如果i未匹配
42         {
43             memset(vis, 0, sizeof(vis));
44             ans += dfs(i);
45         }
46     return ans;
47 }
48 
49 int main()
50 {
51     while (~scanf("%d", &n))
52     {
53         memset(mp, 0, sizeof(mp));
54         for (int i = 1; i <= n; i++)
55         {
56             for (int j = 1; j <= n; j++)
57             {
58                 int v;
59                 scanf("%d", &v);
60                 if (v > 0) mp[i][j] = 1;
61             }
62         }
63         int ans = maxmatch();
64         if (ans < n)printf("-1\n");
65         else
66         {
67             int t = 0;
68             for (int i = 1; i <= n; i++)
69             {
70                 int j;
71                 for (j = 1; j <= n; j++) if (cy[j] == i)break;
72                 if (i != j)
73                 {
74                     stps[t].x = i, stps[t].y = j;
75                     t++;
76                     swap(cy[i], cy[j]);
77                 }
78             }
79             printf("%d\n", t);
80             for (int i = 0; i < t; i++)
81             {
82                 printf("C %d %d\n", stps[i].x, stps[i].y);
83             }
84         }
85     }
86 
87     return 0;
88 }
View Code

5、hdu 2389 Rain on your Parade(最大匹配)(Hopcroft-Carp算法模板)

  题意:有m个人,t时刻后会下雨。地上有n把伞,每把伞只能一个人用。给出人和伞的坐标,以及人的速度,问在下雨前,最多能有多少个人拿到伞?

  思路:人作为x集,伞作为y集,如果人能够到达伞的位置,把该人的编号和该伞的编号连线。之后求最大匹配。Hopcroft-Carp算法。

技术分享
  1 #include<iostream>
  2 #include<queue>
  3 #include<memory.h>
  4 #include<algorithm>
  5 #include<cmath>
  6 using namespace std;
  7 int n,t,m,Nx,Ny,dis;
  8 const int maxn = 3100;//最大行数
  9 const int maxm = 3100;//最大列数
 10 const int maxk = 3100;
 11 const int INF = 0x7fffffff;
 12 bool mp[maxn][maxm];//1表示该ij可以匹配
 13 int cx[maxm];//记录x集合中匹配的y元素是哪一个
 14 int dx[maxm];
 15 int cy[maxn];//记录y集合中匹配的x元素是哪一个
 16 int dy[maxn];
 17 int vis[maxm];//标记该顶点是否访问过
 18 struct point
 19 {
 20     int x;
 21     int y;
 22     int v;
 23 }customer[maxk];
 24 struct point2
 25 {
 26     int x;
 27     int y;
 28 }unbrella[maxk];
 29 bool searchP(void)    //BFS   
 30 {
 31     queue <int> Q;
 32     dis = INF;
 33     memset(dx, -1, sizeof(dx));
 34     memset(dy, -1, sizeof(dy));
 35     for (int i = 1; i <= Nx; i++)
 36         if (cx[i] == -1)
 37         {
 38             Q.push(i); dx[i] = 0;
 39         }
 40     while (!Q.empty())
 41     {
 42         int u = Q.front(); Q.pop();
 43         if (dx[u] > dis) break;        //说明该增广路径长度大于dis还没有结束,等待下一次BFS在扩充  
 44         for (int v = 1; v <= Ny; v++)
 45             if (mp[u][v] && dy[v] == -1)
 46             {        //v是未匹配点  
 47                 dy[v] = dx[u] + 1;
 48                 if (cy[v] == -1) dis = dy[v];    //得到本次BFS的最大遍历层次  
 49                 else
 50                 {
 51                     dx[cy[v]] = dy[v] + 1;         //v是匹配点,继续延伸  
 52                     Q.push(cy[v]);
 53                 }
 54             }
 55     }
 56     return dis != INF;
 57 }
 58 
 59 bool DFS(int u)
 60 {
 61     for (int v = 1; v <= Ny; v++)
 62         if (!vis[v] && mp[u][v] && dy[v] == dx[u] + 1)
 63         {
 64             vis[v] = 1;
 65             if (cy[v] != -1 && dy[v] == dis) continue;   //层次(也就是增广路径的长度)大于本次查找的dis,是searchP被break的情况,也就是还不确定是否是增广路径,只有等再次调用searchP()在判断。  
 66             if (cy[v] == -1 || DFS(cy[v]))
 67             {     //是增广路径,更新匹配集  
 68                 cy[v] = u; cx[u] = v;
 69                 return 1;
 70             }
 71         }
 72     return 0;
 73 }
 74 
 75 int MaxMatch(void)
 76 {
 77     int res = 0;
 78     memset(cx, -1, sizeof(cx));
 79     memset(cy, -1, sizeof(cy));
 80     while (searchP())
 81     {
 82         memset(vis, 0, sizeof(vis));
 83         for (int i = 1; i <= Nx; i++)
 84             if (cx[i] == -1 && DFS(i)) res++;   //查找到一个增广路径,匹配数res++  
 85     }
 86     return res;
 87 }
 88 
 89 int main()
 90 {
 91     int C;
 92     scanf("%d", &C);
 93     int Case = 1;
 94     while (C--)
 95     {
 96         scanf("%d", &t);
 97         memset(mp, 0, sizeof(mp));
 98         scanf("%d", &m);
 99         for (int i = 1; i <= m; i++)
100         {
101             scanf("%d%d%d", &customer[i].x, &customer[i].y, &customer[i].v);
102         }
103         scanf("%d", &n);
104         for (int i = 1; i <= n; i++)
105         {
106             scanf("%d%d", &unbrella[i].x, &unbrella[i].y);
107         }
108         for (int i = 1; i <= n; i++)
109         {
110             for (int j = 1; j <= m; j++)
111             {
112                 double dis = sqrt((unbrella[i].x - customer[j].x)*(unbrella[i].x - customer[j].x) + (unbrella[i].y - customer[j].y)*(unbrella[i].y - customer[j].y));
113                 if (dis <= t*customer[j].v) mp[i][j] = true;
114             }
115         }
116         Nx = n, Ny = m;
117         int ans = MaxMatch();
118         printf("Scenario #%d:\n", Case++);
119         printf("%d\n\n", ans);
120     }
121     return 0;
122 }
View Code

6、hdu 4185 Oil Skimming

  题意:给出一张图,每相邻两个油田可以建立一道沟渠。问最大沟渠的数目。

  思路:给每一个油田编号。油田既作为x集,又作为y集,如果两块油田相邻,则连线。之后求最大匹配。

技术分享
  1 #include<iostream>
  2 #include<queue>
  3 #include<memory.h>
  4 #include<algorithm>
  5 #include<cmath>
  6 using namespace std;
  7 int n, t, m, Nx, Ny, dis;
  8 const int maxn = 650;//最大行数
  9 const int maxm = 650;//最大列数
 10 const int maxk = 360005;
 11 const int INF = 0x7fffffff;
 12 char M[maxn][maxm];
 13 bool mp[maxn][maxn];//1表示该ij可以匹配
 14 int cx[maxk];//记录x集合中匹配的y元素是哪一个
 15 int dx[maxk];
 16 int cy[maxk];//记录y集合中匹配的x元素是哪一个
 17 int dy[maxk];
 18 int vis[maxk];//标记该顶点是否访问过
 19 struct point
 20 {
 21     int x;
 22     int y;
 23 }oils[maxk];
 24 bool searchP(void)    //BFS   
 25 {
 26     queue <int> Q;
 27     dis = INF;
 28     memset(dx, -1, sizeof(dx));
 29     memset(dy, -1, sizeof(dy));
 30     for (int i = 1; i <= Nx; i++)
 31         if (cx[i] == -1)
 32         {
 33             Q.push(i); dx[i] = 0;
 34         }
 35     while (!Q.empty())
 36     {
 37         int u = Q.front(); Q.pop();
 38         if (dx[u] > dis) break;        //说明该增广路径长度大于dis还没有结束,等待下一次BFS在扩充  
 39         for (int v = 1; v <= Ny; v++)
 40             if (mp[u][v] && dy[v] == -1)
 41             {        //v是未匹配点  
 42                 dy[v] = dx[u] + 1;
 43                 if (cy[v] == -1) dis = dy[v];    //得到本次BFS的最大遍历层次  
 44                 else
 45                 {
 46                     dx[cy[v]] = dy[v] + 1;         //v是匹配点,继续延伸  
 47                     Q.push(cy[v]);
 48                 }
 49             }
 50     }
 51     return dis != INF;
 52 }
 53 
 54 bool DFS(int u)
 55 {
 56     for (int v = 1; v <= Ny; v++)
 57         if (!vis[v] && mp[u][v] && dy[v] == dx[u] + 1)
 58         {
 59             vis[v] = 1;
 60             if (cy[v] != -1 && dy[v] == dis) continue;   //层次(也就是增广路径的长度)大于本次查找的dis,是searchP被break的情况,也就是还不确定是否是增广路径,只有等再次调用searchP()在判断。  
 61             if (cy[v] == -1 || DFS(cy[v]))
 62             {     //是增广路径,更新匹配集  
 63                 cy[v] = u; cx[u] = v;
 64                 return 1;
 65             }
 66         }
 67     return 0;
 68 }
 69 
 70 int MaxMatch(void)
 71 {
 72     int res = 0;
 73     memset(cx, -1, sizeof(cx));
 74     memset(cy, -1, sizeof(cy));
 75     while (searchP())
 76     {
 77         memset(vis, 0, sizeof(vis));
 78         for (int i = 1; i <= Nx; i++)
 79             if (cx[i] == -1 && DFS(i)) res++;   //查找到一个增广路径,匹配数res++  
 80     }
 81     return res;
 82 }
 83 
 84 int main()
 85 {
 86     int C,N;
 87     scanf("%d", &C);
 88     int Case = 1;
 89     while (C--)
 90     {
 91         scanf("%d", &N);
 92         memset(mp, 0, sizeof(mp));
 93         int cnt = 0;
 94         for (int i = 1; i <= N; i++)
 95         {
 96             for (int j = 1; j <= N; j++)
 97             {
 98                 cin >> M[i][j];
 99                 if (M[i][j] == #)
100                 {
101                     oils[cnt].x = i;
102                     oils[cnt].y = j;
103                     cnt++;
104                 }
105             }
106         }
107         for (int i = 0; i < cnt; i++)
108         {
109             for (int j = 0; j < cnt; j++)
110             {
111                 if (abs(oils[i].x - oils[j].x) + abs(oils[i].y - oils[j].y) == 1)
112                 {
113                     mp[i+1][j+1] = 1;
114                 }
115             }
116         }
117         Nx = cnt, Ny = cnt;
118         int ans = MaxMatch();
119         printf("Case %d: %d\n", Case++,ans/2);
120     }
121     return 0;
122 }
View Code

7、poj 3020 Antenna Placement(无向二分图的最小路径覆盖)

 

专题之匹配、网络流(一)