Levenshtein算法-比較兩個字元串之間的相似度

package com.sinoup.util;/**
 * Created by Administrator on 2020-4-18.
 */

/**
 * @Title:
 * @ProjectName
 * @Description:比較字元串相似度
 * @author: TongSiYu
 * @date 2020-4-18 14:28
 */

public class StringCompareUtil {


    public static void main(String[] args) {
        //要比較的兩個字元串
        String str1 = "北京北京市朝陽區南平裡11号樓2單元203";
        String str2 = (" 北京 北京市 朝陽區 朝外街 道 ").trim();
        levenshtein(str1, str2);
    }

    /*計算字元串相似度算法——Levenshtein*/
    public static float levenshtein(String str1, String str2) {
        //計算兩個字元串的長度。
        int len1 = str1.length();
        int len2 = str2.length();
        //建立上面說的數組，比字元長度大一個空間
        int[][] dif = new int[len1 + 1][len2 + 1];
        //賦初值，步驟B。
        for (int a = 0; a <= len1; a++) {
            dif[a][0] = a;
        }
        for (int a = 0; a <= len2; a++) {
            dif[0][a] = a;
        }
        //計算兩個字元是否一樣，計算左上的值
        int temp;
        for (int i = 1; i <= len1; i++) {
            for (int j = 1; j <= len2; j++) {
                if (str1.charAt(i - 1) == str2.charAt(j - 1)) {
                    temp = 0;
                } else {
                    temp = 1;
                }
                //取三個值中最小的
                dif[i][j] = min(dif[i - 1][j - 1] + temp, dif[i][j - 1] + 1,
                        dif[i - 1][j] + 1);
            }
        }
        System.out.println("字元串\"" + str1 + "\"與\"" + str2 + "\"的比較");
        //取數組右下角的值，同樣不同位置代表不同字元串的比較
        System.out.println("差異步驟：" + dif[len1][len2]);
        //計算相似度
        float similarity = 1 - (float) dif[len1][len2] / Math.max(str1.length(), str2.length());
        System.out.println("相似度：" + similarity);
        return similarity;
    }

    //得到最小值
    private static int min(int... is) {
        int min = Integer.MAX_VALUE;
        for (int i : is) {
            if (min > i) {
                min = i;
            }
        }
        return min;
    }

}