POJ 1789 求车类型字符串距离最小值 最小生成树 prim算法

题意：车的类型用字符串来描述，两个汽车类型的距离定义为其字符串中不同字符的个数，给定所有的车的类型，求车之间的派生关系，使得总的距离最短，派生关系质量分最大

算法：转化为图论问题，结点对应不同的汽车类型，边的权值就是不同结点字符串距离值，求最小生成树即可（因为题目说明了每个汽车只有一种汽车派生出）

Source Code

Problem: 1789		User: yangliuACMer
Memory: 15688K		Time: 422MS
Language: C++		Result: Accepted

//00:31-0:53 20分钟写完
#include <iostream>
using namespace std;
#define MAXN 2001
#define codeLen 7
#define inf 1000
typedef int elem_t;
int G[MAXN][MAXN];
int pre[MAXN];
char codeArray[MAXN][codeLen];

elem_t prim(int n,elem_t mat[MAXN][MAXN],int* pre){
	elem_t min[MAXN],ret=0;
	int v[MAXN],i,j,k;
	for (i=0;i<n;i++)
		min[i]=inf,v[i]=0,pre[i]=-1;
	for (min[j=0]=0;j<n;j++){
		for (k=-1,i=0;i<n;i++)
			if (!v[i]&&(k==-1||min[i]<min[k]))
				k=i;
			for (v[k]=1,ret+=min[k],i=0;i<n;i++)
				if (!v[i]&&mat[k][i]<min[i])
					min[i]=mat[pre[i]=k][i];
	}
	return ret;
}

int computeDis(char s1[], char s2[]){
	int k = 0;
	for(int i = 0; i < codeLen; i++){
		if(s1[i] != s2[i]){
			k++;
		}
	}
	return k;
}

int main(){
	int n, i, j;
	while(cin>>n,n){
		for(i = 0; i < n; i++){
			cin>>codeArray[i];
		}
		for(i = 0; i < n; i++){
			G[i][i] = 0;//对角线为0
			for(j = i+1; j < n; j++){
				G[i][j] = G[j][i] = computeDis(codeArray[i] , codeArray[j]);
			}
		}
		int result =  prim(n, G, pre);
		cout<<"The highest possible quality is 1/"<<result<<"."<<endl;
	}
	return 0;
}