【深度优先搜索实战】力扣547题:省份数量问题的图论解法

题目解读
我们面对的是一个典型的图论问题:给定一个城市的连接矩阵,需要计算其中相互连通的城市群(省份)数量。这个问题可以抽象为无向图中的连通分量计算,每个城市代表图中的一个节点,城市之间的连接关系代表图中的边。题目要求我们找出这些节点形成了多少个互不连通的子图,这在实际应用中可以用来解决社交网络中的朋友圈划分、计算机网络中的连通区域等问题。
解题思路
使用深度优先搜索(DFS)的策略,首先将邻接矩阵转换为邻接表的形式存储,这样可以更高效地进行图遍历。然后初始化一个标记数组来记录哪些节点已经被访问过。对于每个未被访问的节点,执行DFS遍历,将所有与之相连的节点标记为已访问。每次从一个新节点开始的DFS遍历就对应着一个新的连通分量(省份)。最后统计进行了多少次这样的DFS遍历,就是所求的省份数量。
代码注释
class Solution {
public:
vector<int> edge[201]; // 邻接表存储图的连接关系
int flag[201]={0}; // 标记数组,记录节点是否被访问过
int n; // 城市总数
// 深度优先搜索函数
void delconnect(int now){
if(flag[now])return; // 如果已经访问过则返回
flag[now]=true; // 标记当前节点为已访问
for(int i=0;i<edge[now].size();i++){
delconnect(edge[now][i]); // 递归访问所有相邻节点
}
}
int findCircleNum(vector<vector<int>>& isConnected) {
int ret=0; // 省份计数器
n=isConnected.size(); // 获取城市数量
// 将邻接矩阵转换为邻接表
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(isConnected[i][j])
edge[i].push_back(j);
}
}
// 遍历所有城市,统计连通分量
for(int i=0;i<n;i++){
if(!flag[i]){ // 如果该城市未被访问过
delconnect(i); // 执行DFS遍历
ret++; // 增加省份计数
}
}
return ret;
}
};原创内容 转载请注明出处




