等价类问题使用并查集解决即可
题目要求所谓的移动只能在一个字符串的奇数号字符之间或者偶数号字符之间进行,所以我们要分别统计两个字符串的奇数号字符和偶数号字符的分布,然后比较是否等价,最基本的并查集操作
class Solution {
public:
int find(int ufs[],int y)
{
return ufs[y]<0?y:find(ufs,ufs[y]);
}
void setUnion(int ufs[],int r1,int r2)
{
if(r1==r2) return;
if(ufs[r1]<ufs[r2])
{
ufs[r1]+=ufs[r2];
ufs[r2]=r1;
}
else
{
ufs[r2]+=ufs[r1];
ufs[r1]=r2;
}
}
bool judge(int a[][2],int b[][2])
{
for(int i=0;i<26;i++)
if(a[i][0]!=b[i][0]||a[i][1]!=b[i][1])
return false;
return true;
}
int numSpecialEquivGroups(vector<string>& A) {
int n=A.size(),ufs[1000],ans=0,cnt[1000][26][2]={};
memset(ufs,-1,sizeof(int)*n);
for(int i=0;i<n;i++)
for(int j=0;j<A[i].length();j++)
cnt[i][A[i][j]-'a'][j%2]++;//提前打好表,加快查询速度
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
if(judge(cnt[i],cnt[j]))
setUnion(ufs,find(ufs,i),find(ufs,j));
for(int i=0;i<n;i++) if(ufs[i]<0) ans++;
return ans;
}
};