面试算法117:相似的字符串

题目

如果交换字符串X中的两个字符就能得到字符串Y,那么两个字符串X和Y相似。例如,字符串"tars"和"rats"相似(交换下标为0和2的两个字符)、字符串"rats"和"arts"相似(交换下标为0和1的字符),但字符串"star"和"tars"不相似。

输入一个字符串数组,根据字符串的相似性分组,请问能把输入数组分成几组?如果一个字符串至少和一组字符串中的一个相似,那么它就可以放到该组中。假设输入数组中的所有字符串的长度相同并且两两互为变位词。例如,输入数组为["tars","rats","arts","star"],可以分成两组,一组为{"tars","rats","arts"},另一组为{"star"}。

分析

把输入数组中的每个字符串看成图中的一个节点。如果两个字符串相似,那么它们对应的节点之间有一条边相连,也就属于同一个子图。例如,字符串["tars","rats","arts","star"]根据相似性分别属于两个子图

java 复制代码
public class Test {
    public static void main(String[] args) {
        String[] A = {"tars", "rats", "arts", "star"};
        int result = numSimilarGroups(A);
        System.out.println(result);
    }

    public static int numSimilarGroups(String[] A) {
        int[] fathers = new int[A.length];
        for (int i = 0; i < fathers.length; i++) {
            fathers[i] = i;
        }

        int groups = A.length;
        for (int i = 0; i < A.length; i++) {
            for (int j = i + 1; j < A.length; j++) {
                if (similar(A[i], A[j]) && union(fathers, i, j)) {
                    groups--;
                }
            }
        }

        return groups;
    }

    private static boolean similar(String str1, String str2) {
        int diffCount = 0;
        for (int i = 0; i < str1.length(); i++) {
            if (str1.charAt(i) != str2.charAt(i)) {
                diffCount++;
            }
        }

        return diffCount <= 2;
    }

    private static boolean union(int[] fathers, int i, int j) {
        int fatherOfI = findFather(fathers, i);
        int fatherOfJ = findFather(fathers, j);
        if (fatherOfI != fatherOfJ) {
            fathers[fatherOfI] = fatherOfJ;
            return true;
        }

        return false;
    }

    private static int findFather(int[] fathers, int i) {
        if (fathers[i] != i) {
            fathers[i] = findFather(fathers, fathers[i]);
        }

        return fathers[i];
    }

}
相关推荐
白水先森2 小时前
C语言作用域与数组详解
java·数据结构·算法
想唱rap2 小时前
直接选择排序、堆排序、冒泡排序
c语言·数据结构·笔记·算法·新浪微博
老葱头蒸鸡3 小时前
(27)APS.NET Core8.0 堆栈原理通俗理解
算法
视睿3 小时前
【C++练习】06.输出100以内的所有素数
开发语言·c++·算法·机器人·无人机
柠檬07114 小时前
matlab cell 数据转换及记录
算法
YuTaoShao4 小时前
【LeetCode 每日一题】2221. 数组的三角和
数据结构·算法·leetcode
little~钰5 小时前
树上倍增和LCA算法---上
算法
聪明的笨猪猪5 小时前
Java SE “JDK1.8新特性”面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
力扣蓝精灵5 小时前
今日分享 整数二分
算法
mc23565 小时前
5分钟学会微算法——Brian Kernighan 算法
算法