3.3_计算字符串的相似度
和计算两字符串的最长公共子序列相似。
设Ai为字符串A(a1a2a3 … am)的前i个字符(即为a1,a2,a3 … ai)
设Bj为字符串B(b1b2b3 … bn)的前j个字符(即为b1,b2,b3 … bj)
设 L(i , j)为使两个字符串和Ai和Bj相等的最小操作次数。
当ai等于bj时 显然L(i, j)=L(i-1, j-1)
当ai不等于bj时
若将它们修改为相等,则对两个字符串至少还要操作L(i-1, j-1)次
若删除ai或在Bj后添加ai,则对两个字符串至少还要操作L(i-1, j)次
若删除bj或在Ai后添加bj,则对两个字符串至少还要操作L(i, j-1)次
此时L(i, j)=min( L(i-1, j-1), L(i-1, j), L(i, j-1) ) + 1
显然,L(i, 0)=i,L(0, j)=j, 再利用上述的递推公式,可以直接计算出L(i, j)值。
为了保持与书中代码一致,下面的函数参数类型是string,而不是char*。
int string_distance(const string& sa, const string& sb) { const int sz_a=sa.size()+1; const int sz_b=sb.size()+1; int i,j,k,tmp; vector< vector<int> > arr(sz_a, vector<int>(sz_b) ); for (i=0; i<sz_a; ++i) arr[i][0]=i; for (j=0; j<sz_b; ++j) arr[0][j]=j; for (i=1; i<sz_a; ++i){ for (j=1; j<sz_b; ++j){ if( sa[i-1] == sb[j-1]) arr[i][j]=arr[i-1][j-1]; else{ tmp= arr[i-1][j] > arr[i][j-1] ? arr[i][j-1] : arr[i-1][j]; if (tmp>arr[i-1][j-1]) tmp=arr[i-1][j-1]; arr[i][j]=tmp+1; } } } return arr[sz_a-1][sz_b-1]; }
由于只要求计算两字串的距离,计算时,只用到两列数据,因而可以对代码进一步优化,节省空间。
int string_distance2(const string& sa, const string& sb) { const int sz_a=sa.size()+1; const int sz_b=sb.size()+1; int sz_max=sz_a; int sz_min=sz_b; const char *longer=sa.data(); const char *shorter=sb.data(); if (sz_a < sz_b){ sz_max=sz_b; sz_min=sz_a; longer=sb.data(); shorter=sa.data(); } int i,j,k,tmp; vector<int> arr(sz_min+1); for (j=0; j<sz_min; ++j) arr[j+1]=j; for (i=1; i<sz_max; ++i){ arr[0]=i; for (j=1; j<sz_min; ++j){ if( longer[i-1] != shorter[j-1]) { tmp= arr[j+1] > arr[j] ? arr[j] : arr[j+1]; if (tmp>arr[j-1]) tmp=arr[j-1]; arr[j]=tmp+1; //tmp= min(arr[j-1],arr[j],arr[j+1]) } } for(j=sz_min-1; j>=0; –j) arr[j+1]=arr[j]; } return arr[sz_min]; }
上面的代码还可进一步优化,比如通过指针而不是数组名来访问内存。如果内存足够大,可以多申请空间,每次循环,通过修改保存的数据起始位置,避免内存复制。