以下面的题为例
有三门课程,三个学生,连线表示此学生选择了这么课程,只有选择了此门课程才能代表这么课程,最大匹配的结果 会使得最多的课程被学生代表。
课程作为一个集合 学生作为一个集合 有如下关系
则,采用DFS 匈牙利算法,计算是否没门课程都有学生代表的代码如下
#include<stdio.h>
#include<string>
#define M 301
int P,N;
bool course[M][M],used[M];
int match[M];
bool DFS(int k)
{
int i,temp;
for(i=1;i<=N;i++)
{
if(course[k][i] && !used[i])
{
used[i]=1;
temp=match[i];
match[i]=k;
if(temp==-1 || DFS(temp))
return 1;
match[i]=temp;
}
}
return 0;
}
int MaxMatch()
{
int i,MatchNum=0;
bool a;
memset(match,-1,sizeof(match));
for(i=1;i<=P;i++)
{
memset(used,0,sizeof(used));
a=DFS(i);
if(a)
MatchNum++;
if(MatchNum==P) break;
}
return MatchNum;
}
int main()
{
int i,j,num,t,n;
scanf("%d %d",&P,&N);
for(i=1;i<=P;i++)
{
scanf("%d",&num);
for(j=0;j<num;j++)
{
scanf("%d",&t);
course[i][t]=1;
}
}
if(MaxMatch()==P)
printf("YES\n");
else
printf("No\n");
system("PAUSE");
return 0;
}
对代码的过程做详细分析,使过程更清晰
输入的数据后 course数组变为 course[i][j]表示i行j列 表示第i个课程被第j个学生选择
1 | 1 | 1 |
1 | 1 | 0 |
1 | 0 | 0 |
在MaxMatch()函数中第一次调用DFS函数
DFS(1)
在开始处 k=1 used match 的状态为
used
0 | 0 | 0 |
match
-1 | -1 | -1 |
在if 语句执行时 used[1]被标志 match[1]=1 表示第一个学生代表第一门课
used 和match的状态分别变为
used
1 | 0 | 0 |
match
1 | -1 | -1 |
temp的值为 -1 调用退出
在MaxMatch()函数中第二次调用DFS函数
DFS(2)
在开始处 k=2 used match 的状态为
used
0 | 0 | 0 |
match
1 | -1 | -1 |
DFS函数中for循环的第一层
在if 语句执行时 used[1] 被标志 temp=1 match[2]=1;
由于课程2与学生1匹配,递归调用DFS(1)为课程1寻找新的匹配学生
这时used[1] 已经被标记
继续判断used[2] 符合条件 match[2] 依然为-1 匹配成工 向调用函数返回1
在MaxMatch()函数中第三次调用DFS函数
DFS(3)
在开始处 k=3 used match 的状态为
used
0 | 0 | 0 |
match
2 | 1 | -1 |
DFS函数中for循环的第一层
在if 语句执行时 used[1] 被标志 temp=2 match[3]=1;
由于课程3与学生1匹配,递归调用DFS(2)为课程2寻找新的匹配学生
这时used[1] 已经被标记
继续判断used[2] 符合条件 此时 match[2] =1 即第二名学生已经和第一门课匹配 递归调用DFS(1)为第一门课寻找新的学生
由于1 和2 已经被标记 找到学生3 和第一门课程匹配
DFS(1)返回后 第二门课程可以使用刚刚第一门课程不再使用的学生2 match[2]=2;
DFS(2)返回后 让出第一个学生 第一名们课程使用
至此没门课都有自己的代表学生。