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

来源:互联网 发布:mac appstore 太慢 编辑:程序博客网 时间:2024/05/17 03:07

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

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

Source Code

Problem: 1789 User: yangliuACMerMemory: 15688K Time: 422MSLanguage: C++ Result: Accepted
//00:31-0:53 20分钟写完#include <iostream>using namespace std;#define MAXN 2001#define codeLen 7#define inf 1000typedef 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;//对角线为0for(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;}