839. Similar String Groups

问题：

给定一组字符串，若其中两个字符串中，其中一个字符串任意两个字符互换后=另一个字符串，那么说这两个字符串相似。

求这组字符串中的相似字符串组，有多少个。

Example 1:
Input: A = ["tars","rats","arts","star"]
Output: 2

Constraints:
1 <= A.length <= 2000
1 <= A[i].length <= 1000
A.length * A[i].length <= 20000
All words in A consist of lowercase letters only.
All words in A have the same length and are anagrams of each other.
The judging time limit has been increased for this question.

 

解法：并查集（Disjoint Set）

将index作为对象，在DisjointSet中进行操作。

相似的算法：

    bool isSimilar(string& a, string& b) {
        int n=0;
        for(int i=0; i<a.length(); i++) {
            if(a[i]!=b[i] && ++n>2) return false;
        }
        return true;
    }

其中不相同的字符超过2个，则不相似。

 

代码参考：

class Solution {
public:
    bool isSimilar(string& a, string& b) {
        int n=0;
        for(int i=0; i<a.length(); i++) {
            if(a[i]!=b[i] && ++n>2) return false;
        }
        return true;
    }
    int numSimilarGroups(vector<string>& A) {
        DisjointSet DS(A.size());
        for(int i=1; i<A.size(); i++) {
            for(int j=0; j<i; j++) {
                if(isSimilar(A[i], A[j])) {
                    DS.merge(i, j);
                }
            }
        }
        return DS.getGroupCount();
    }
};

 

Disjoint Set类的代码参考：

class DisjointSet {
public:
    DisjointSet(int n):root(n, 0), rank(n, 0) {
        for(int i=0; i<n; i++) {
            root[i] = i;
        }
    }
    int find(int i) {
        if(i != root[i]) {
            root[i] = find(root[i]);
        }
        return root[i];
    }
    bool merge(int x, int y) {
        int x_root = find(x);
        int y_root = find(y);
        if(x_root == y_root) return false;
        if(rank[x_root] > rank[y_root]) {
            root[y_root] = x_root;
        } else if(rank[y_root] > rank[x_root]) {
            root[x_root] = y_root;
        } else {
            root[x_root] = y_root;
            rank[y_root]++;
        }
        return true;
    }
    int getGroupCount() {
        int res = 0;
        for(int i=0; i<root.size(); i++) {
            if(root[i] == i) res++;
        }
        return res;
    }
private:
    vector<int> root;
    vector<int> rank;
};