1. 程式人生 > 其它 >LeetCode每日一題:839. 相似字串組

LeetCode每日一題:839. 相似字串組

技術標籤:leetcode字串

分析

本題考查並查集基本操作。
從前到後把每個字串與在其之後的字串進行比較,如果二者有兩個對應位置上的字元不同的話就合併其所在的集合。
最後所有父親是其自身的點的總數就是答案。

C++ 程式碼

class Solution {
public:
    int n,m,ans;
    vector<int> fa;
    int find(int x)
    {
        if(fa[x]!=x) fa[x]=find(fa[x]);
        return fa[x];
    }
    bool check(int a,int b,vector<string>& s)   //判斷兩個字串是否只有兩個相對位置的不同
    {
        int co=0,idx1=-1,idx2=-1;
        for(int i=0;i<m;i++)
        {
            if(s[a][i]!=s[b][i])
            {
                co++;
                if(co==1) idx1=i;
                else if(co==2) idx2=i;
                else    
                    return false;       //有兩個以上的位置不同,返回false
            }
        }
        if(co==0) return true;          //字串完全相同,返回true
        if(co==1) return false;         //只有一個位置不同,返回false
        if(s[a][idx1]==s[b][idx2] && s[a][idx2]==s[b][idx1])    //滿足題意,返回true
            return true;
        return false;
    }
    int numSimilarGroups(vector<string>& strs) {
        n=strs.size(),m=strs[0].size();
        fa.resize(n);
        for(int i=0;i<n;i++)
        {
            fa[i]=i;
        }
        for(int i=0;i<n;i++)
        {
            for(int j=i+1;j<n;j++)
            {
                int faa=find(i),fbb=find(j);
                if(faa==fbb) continue;  
                if(check(i,j,strs)) 
                {
                    fa[faa]=fbb;
                }
            }
        }
        for(int i=0;i<n;i++)
        {
            if(i==find(i))      //此時為1個大集合
                ans++;
        }
        return ans;
    }
};