该题可用dfs、bfs、并查集。我采用的是dfs,方便快捷。
虽然dfs写起来很简单,但是也需要优化一下时间复杂度,在dfs遍历所有点的时候,可以if(x!=i)去除对自己进行的过程,因为自己和自己本来就是朋友,M[I][I]恒为一,所以不需要判断,只需要判断自己和他人是否形成一个朋友圈。
class Solution {
public:
int findCircleNum(vector<vector>& M) {
int count=0;
int n=M.size();
int visited[n];
for(int i=0;i<n;i++) visited[i]=0;
for(int i=0;i<n;i++){
//cout<<visited[i]<<endl;
if(visited[i]==0){
// cout<<“run”;
dfs(visited,M,i);
count++;
}
}
return count;
}
void dfs(int visited[],vector<vector>& M,int x){
//cout<<x<<endl;
for(int i=0;i<M.size();i++){
if(x!=i){
if(visited[i]==0&&M[x][i]==1){
visited[i]=1;
dfs(visited,M,i);
}
}
}
}
};