很明显的一个隐式图下连通分量计数。下面是比较明显的一个思路,通过遍历所有点,对于以前没有访问过的点进行深度优先搜索找到该连通分量内的所有结点,并在一个数组内进行标记。整体来说是O(n^2)级别,比25%左右的C++代码快。
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <fstream>
#include <queue>
using namespace std;
class Solution{
public:
bool judgeEdge(string s1,string s2)
{
int idx1 = -1,idx2 = -1;
if(s1.size()!=s2.size())
return false;
for(int i = 0 ;i<s1.size();i++)
{
if(s1[i]!=s2[i])
{
if(idx1==-1)
idx1 = i;
else if(idx2==-1)
idx2 = i;
else
return false;
}
}
if(s1[idx1]==s2[idx2] &&s1[idx2]==s2[idx1])
{
return true;
}
else
return false;
}
int numSimilarGroups(vector<string>& A) {
/*
隐式图的连通分量搜索
对隐式图的所有节点进行遍历,找寻它们之间是否能够连通。如果能够连通,则加1,并加入到一个并查集中。
*/
vector<bool> isVisit(A.size(),false);
int counting = 0;
int truePoint = 0;
for(int i = 0;i<A.size();i++)
{
if(isVisit[i])
continue;
cout<<"Point:"<<i<<endl;
isVisit[i] = true;
truePoint++;
counting++;
queue<int> q;
q.push(i);
while(!q.empty())
{
int index = q.front(); q.pop();
for(int j = i+1;j<A.size();j++)
{
if(isVisit[j])
continue;
if(judgeEdge(A[index],A[j]))
{
isVisit[j] = true;
truePoint++;
if(truePoint == A.size())
return counting;
q.push(j);
}
}
}
}
return counting;
}
};
int main(void)
{
Solution s;
vector<string> test{"tars","rats","arts","star"};
cout<<s.numSimilarGroups(test)<<endl;
return 0;
}