您的位置:首页 > 其它

POJ 1469 COURSES (二分匹配,邻接表)

2014-04-23 20:42 429 查看
http://poj.org/problem?id=1469

#include<stdio.h>
#include<string.h>
#include<iostream>
#define point_MAX 10000
#define edge_MAX 100000
using namespace std;
struct EDGE
{
int to;/*指向的点*/
int next;/*指向的下一条邻边*/
int w;/*权值*/
}edge[edge_MAX];
int len;/*边的数量*/
int point[point_MAX];
int nx,ny;/*x集合和y集合中顶点的个数*/
int cx[point_MAX],cy[point_MAX];/*用来记录x集合中匹配的y元素是哪个!*/
int visited[point_MAX];/*用来记录该顶点是否被访问过!*/
void init()/*初始化*/
{
len=0;
memset(point,0,sizeof(point));
}
int add_edge(int a,int b,int w)/*添加由a指向b的权值为w的边*/
{
int i;
for(i=point[a];i;i=edge[i].next)
{
if(edge[i].to==b)
{
if(edge[i].w<w)/*重边取大*/
{
edge[i].w=w;
return 1;/*有重边,并覆盖*/
}
return -1;/*有重边舍去*/
}
}
/*以上处理重边*/
len++;
edge[len].w=w;
edge[len].to=b;
edge[len].next=point[a];
point[a]=len;
return 0;/*无重边,插入*/
}
int path(int u)/*由点u开始寻找增广路,即将u与刚刚的连线去掉,继而寻找新的连线*/
{
int v;
for(v=point[u];v;v=edge[v].next)
{
if(!visited[edge[v].to])/*找到的这个必须之前没有找过的点,就是在不与其他点冲突的情况下寻找增广路*/
{
visited[edge[v].to]=1;
if(cy[edge[v].to]==-1||path(cy[edge[v].to]))/*如果y集合中的v元素没有匹配或者是v已经匹配,但是从cy[v]中能够找到一条增广路*/
{
/*cx[u]=edge[v].to;这是无向图中才要的*/
cy[edge[v].to]=u;
return 1;
}
}
}
return 0;
}
int maxmatch()/*遍历所有的点,不断寻找增广路,直到结束*/
{
int res=0,i;
for(i=0;i<point_MAX;i++)
cx[i]=cy[i]=-1;/*初始值为-1表示两个集合中都没有匹配的元素!*/
for(i=0;i<=nx;i++)
{
if(cx[i]==-1)
{
memset(visited,0,sizeof(visited));
res+=path(i);/*寻找增广路*/
}
}
return res;
}
int main()
{
int i,n,j,x,y,ans,t;
cin>>t;
while(t--)
{
scanf("%d%d",&nx,&ny);//nx-->course,ny-->student
//ny=nx;
init();/*初始化*/
for(i=0;i<nx;i++)
{
scanf("%d",&n);
for(j=0;j<n;j++)
{
scanf("%d",&y);
add_edge(i+1,y,1);
}
}
ans=maxmatch();/*匈牙利算法*/
/*for(i=0;i<ny;i++)
{
if(cy[i]==-1)break;
}*/
if(ans==nx)
printf("YES\n");
else cout<<"NO"<<endl;
}
return 0;
}


View Code
模板里的数从0开始。

匈牙利算法每次寻找增广路的结束条件是能够递归找到y集合中的点没有被x集合中的点连接的。此时即可返回1.表示可增广。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: