hdu3722 Card Game (KM)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3722

题解：将字符串分成X和Y集合(每个集合都是这n个节点),对于i和j这两个节点,他们的权值weight[i][j]就是将字符串s[i]拼接在字符串s[j]后面所得的值,这样建立起带权值的二分图之后,然后就是利用KM算法求解该二分图的最大权匹配.

#include <stdio.h>      
#include <string.h>     
#include <math.h>    
#define INF 0x3f3f3f3f  
#define MAXN 201 

int w[MAXN][MAXN],match[MAXN],n;    
int lx[MAXN],ly[MAXN],slack[MAXN];    
int visitx[MAXN],visity[MAXN]; 
char str[MAXN][1001];

int find(int x)  
{  
	int i,temp;  
	visitx[x]=1;  
	for(i=1;i<=n;++i)  
	{  
		if(visity[i])  
			continue;  
		temp=lx[x]+ly[i]-w[x][i];  
		if(temp==0)  
		{  
			visity[i]=1;  
			if(match[i]==-1||find(match[i]))  
			{  
				match[i]=x;//找到增广轨  
				return 1;  
			}  
		}  
		else if(slack[i]>temp)  
		{//不在相等子图中slack 取最小的  
			slack[i]=temp;  
		}  
	}  
	return 0;  
}  

void KM()  
{  
	int i,j,d;  
	memset(ly,0,sizeof(ly));  
	memset(match,-1,sizeof(match));  
	for(i=1;i<=n;++i)  
	{//lx初始化为与它关联边中最大的   
		lx[i]=w[i][1];  
		for(j=2;j<=n;++j)  
			if(w[i][j]>lx[i])  
				lx[i]=w[i][j];  
	}  
	for(i=1;i<=n;++i)  
	{  
		for(j=1;j<=n;++j)  
			slack[j]=INF;  
		while(1)  
		{  
			memset(visitx,0,sizeof(visitx));  
			memset(visity,0,sizeof(visity));  
			if(find(i))  
				break;  
			d=INF;  
			for(j=1;j<=n;++j)  
			{  
				if(!visity[j]&&d>slack[j])  
					d=slack[j];  
			}  
			for(j=1;j<=n;++j)  
			{  
				if(visitx[j])  
					lx[j]-=d;  
				if(visity[j])  
					ly[j]+=d;  
				else  
					slack[j]-=d;  
			}  
		}  
	}  
}  

int getWeight(int i,int j)
{
	int len1,len2,x,y;
	len1=strlen(str[i]);
	len2=strlen(str[j]);
	x=0,y=len1-1;
	while(x<len2&&y>=0&&str[i][y]==str[j][x])
	{
		x++;
		y--;
	}
	return x;
}

int main()
{
	int i,j,ans;
	while(scanf("%d",&n)!=EOF)
	{
		getchar();
		for(i=1;i<=n;++i)
			gets(str[i]);
			//scanf("%s",str[i]);
		for(i=1;i<=n;++i)
		{
			for(j=1;j<=n;++j)
			{
				if(i!=j)
				    w[i][j]=getWeight(i,j);
				else
					w[i][j]=0;
			}
		}
		KM();
		ans=0;
		for(i=1;i<=n;++i)
		{
			if(match[i]!=-1)
			   ans+=w[match[i]][i];
		}
		printf("%d\n",ans);
	}
	return 0;
}