LeetCode每日一題:839. 相似字串組
阿新 • • 發佈:2021-02-01
分析
本題考查並查集基本操作。
從前到後把每個字串與在其之後的字串進行比較,如果二者有兩個對應位置上的字元不同的話就合併其所在的集合。
最後所有父親是其自身的點的總數就是答案。
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; } };