个人练习-Leetcode-839. Similar String Groups

最新推荐文章于 2024-07-25 11:15:54 发布

Rstln

最新推荐文章于 2024-07-25 11:15:54 发布

阅读量47

点赞数

文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/Rstln/article/details/129969553

版权

该文章介绍了一个LeetCode问题，涉及字符串相似性和并查集数据结构。通过判断两个字符串是否可以通过交换两个位置的字符变得相等，将相似字符串归为一组。使用并查集解决此问题，计算连通分量的数量，即相似组的个数。

摘要由CSDN通过智能技术生成

题目链接：https://leetcode.cn/problems/similar-string-groups/

题目大意：如果有两个字符串，其中一个字符串交换且仅交换两个位置的字符就和第二个字符串相等，那么这两个字符串就是相似的。当然，如果两个字符串相等，那它们也是相似的。两个相似的字符串可以并入一个相似组。注意，如果s1与s2相似，s2与s3相似，s1与s3不相似，那么它们三个字符串也算在一个相似组内。

给出一系列字符串strs[]，保证里面的字符串长度相同并且每一个字符串都是其他字符串的1个字母异位词。求里面有多少个相似组。

思路：题目描述有点怪，一直没看懂意思，对strs[]里的组成也不明白。看了题解后才知道可以转换成并查集做，因为【相似组】的并入规则和并查集是一致的。如果把一个字符串看成一个节点，那么相似就是两个节点之间有边连接，题意就变成了求连通分量的个数。

因此构建并查集来做就好，最后数root（每个并查集的根）的个数就好。

完整代码

class Solution {
public:
    bool isSim(string& s1, string& s2) {
        int cnt = 0;
        for (int i = 0; i < s1.length(); i++) {
            if (s1[i] != s2[i]) {
                cnt++;
                if (cnt > 2)
                    return false;
            }
        }
        return true;
    }

    int findRoot(vector<int> fa, int x) {
        if (fa[x] == -1)
            return x;
        else
            return findRoot(fa, fa[x]);
    }


    int numSimilarGroups(vector<string>& strs) {
        int N = strs.size();
        vector<int> fa(N, -1);

        for (int i = 0; i < N; i++) {
            for (int j = i+1; j < N; j++) {
                int fi = findRoot(fa, i), fj = findRoot(fa, j);
                if (fi == fj)
                    continue;
                if (isSim(strs[i], strs[j]))                     
                    fa[j] = fa[fj] = fi;
            }
        }

        for (int i = 0; i < N; i++) {
            if (fa[i] != -1)
                fa[i] = findRoot(fa, i);
        }

        int ret = 0;
        for (int i = 0; i < N; i++) {
            if (fa[i] == -1)
                ret++;
        }

        return ret;
    }
};