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; }