c++ 深度和广度优先搜索

深度优先搜索 (DFS)

DFS 是一种递归的遍历方法,它从根节点开始,沿着树的每一条路径深度遍历,直到到达叶子节点,之后回溯到上一个节点继续遍历。

主要特点:

  • 深度优先搜索:从根节点开始,一直向下搜索直到叶子节点,然后回溯。
  • 适合递归实现,栈可以用来模拟递归过程。
  • 在树或图中,DFS的复杂度通常为 O(V + E),其中V是节点数,E是边数。
#include <iostream>
#include <vector>

using namespace std;

// 定义二叉树节点
struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

// 递归实现深度优先遍历(前序遍历)
void dfs(TreeNode* root) {
    if (!root) {
        return;  // 递归结束条件
    }
    
    // 访问当前节点
    cout << root->val << " ";
    
    // 递归访问左子树
    dfs(root->left);
    
    // 递归访问右子树
    dfs(root->right);
}

int main() {
    // 构建一个简单的二叉树
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    // 执行深度优先遍历(前序遍历)
    cout << "DFS Preorder Traversal: ";
    dfs(root);
    cout << endl;

    return 0;
}

图的连通分量

DFS可以用来在无向图中查找连通分量。一个连通分量是图中一个子集,其中每一对节点都通过边相互连接。DFS可以遍历每个连通分量,从一个未访问的节点开始遍历并标记所有可以访问的节点,直到整个连通分量都被遍历到。

示例:计算图中的连通分量

#include <iostream>
#include <vector>

using namespace std;

class Graph {
public:
    int V;  // 图的节点数
    vector<vector<int>> adj;  // 邻接表表示的图

    Graph(int V) {
        this->V = V;
        adj.resize(V);
    }

    void addEdge(int u, int v) {
        adj[u].push_back(v);
        adj[v].push_back(u);  // 无向图
    }

    void dfs(int v, vector<bool>& visited) {
        visited[v] = true;
        cout << v << " ";  // 访问当前节点

        // 遍历所有邻居
        for (int neighbor : adj[v]) {
            if (!visited[neighbor]) {
                dfs(neighbor, visited);
            }
        }
    }

    int countConnectedComponents() {
        vector<bool> visited(V, false);
        int count = 0;

        // 对每一个未访问过的节点执行DFS
        for (int i = 0; i < V; ++i) {
            if (!visited[i]) {
                cout << "Connected component " << ++count << ": ";
                dfs(i, visited);
                cout << endl;
            }
        }

        return count;
    }
};

int main() {
    Graph g(6);  // 创建一个有6个节点的图

    // 添加边
    g.addEdge(0, 1);
    g.addEdge(0, 2);
    g.addEdge(3, 4);
    g.addEdge(5, 5);  // 自环,表示一个单独的连通分量

    cout << "Number of connected components: " << g.countConnectedComponents() << endl;

    return 0;
}

广度优先搜索 (BFS)

BFS 是一种逐层遍历的算法,从根节点开始,先访问根节点的所有邻居节点,再依次访问其邻居的邻居,直到所有节点都被访问。

主要特点:

  • 广度优先搜索:从根节点开始,逐层访问所有节点。
  • BFS 使用队列来存储当前层的节点,先进先出 (FIFO) 的队列实现了逐层遍历。
  • 在树或图中,BFS的复杂度通常为 O(V + E),其中V是节点数,E是边数。
#include <iostream>
#include <queue>
#include <vector>

using namespace std;

// 定义二叉树节点
struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

// 广度优先遍历(层序遍历)
vector<vector<int>> bfs(TreeNode* root) {
    vector<vector<int>> result;  // 存储每一层的节点
    if (!root) {
        return result;
    }
    
    queue<TreeNode*> q;  // 队列
    q.push(root);

    while (!q.empty()) {
        int levelSize = q.size();  // 当前层的节点数
        vector<int> currentLevel;  // 存储当前层的节点值

        for (int i = 0; i < levelSize; ++i) {
            TreeNode* node = q.front();
            q.pop();
            currentLevel.push_back(node->val);

            // 将左右子节点加入队列
            if (node->left) {
                q.push(node->left);
            }
            if (node->right) {
                q.push(node->right);
            }
        }
        result.push_back(currentLevel);  // 当前层遍历完毕
    }

    return result;
}

int main() {
    // 构建一个简单的二叉树
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    // 执行广度优先遍历(层序遍历)
    vector<vector<int>> result = bfs(root);

    cout << "BFS Level Order Traversal: \n";
    for (const auto& level : result) {
        for (int val : level) {
            cout << val << " ";
        }
        cout << endl;
    }

    return 0;
}

最短路径问题

BFS非常适合解决无权图中的最短路径问题,即在图中寻找从起点到终点的最短路径。由于BFS是逐层扩展的,它可以保证第一次到达某个节点时就是从起点到该节点的最短路径。

示例:无权图中从源节点到目标节点的最短路径

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

class Graph {
public:
    int V;  // 节点数量
    vector<vector<int>> adj;  // 邻接表表示的图

    Graph(int V) {
        this->V = V;
        adj.resize(V);
    }

    void addEdge(int u, int v) {
        adj[u].push_back(v);
        adj[v].push_back(u);  // 无向图
    }

    // 广度优先搜索,计算从source到其他节点的最短路径
    vector<int> bfsShortestPath(int source) {
        vector<int> dist(V, -1);  // dist[i]表示从source到节点i的距离
        queue<int> q;

        // 初始化
        dist[source] = 0;
        q.push(source);

        while (!q.empty()) {
            int node = q.front();
            q.pop();

            // 遍历所有邻居
            for (int neighbor : adj[node]) {
                if (dist[neighbor] == -1) {  // 如果该邻居未被访问
                    dist[neighbor] = dist[node] + 1;
                    q.push(neighbor);
                }
            }
        }

        return dist;
    }
};

int main() {
    Graph g(6);

    // 添加边
    g.addEdge(0, 1);
    g.addEdge(0, 2);
    g.addEdge(1, 3);
    g.addEdge(1, 4);
    g.addEdge(2, 5);

    vector<int> dist = g.bfsShortestPath(0);  // 从节点0开始BFS

    cout << "Shortest path distances from node 0: ";
    for (int i = 0; i < dist.size(); ++i) {
        cout << dist[i] << " ";
    }
    cout << endl;

    return 0;
}

拓扑排序

拓扑排序是有向无环图(DAG)中的一种排序方式,它将图中的所有节点排列成一个线性序列,且每条边 (u, v) 都满足 uv 前面。拓扑排序有多个应用,例如任务调度、编译依赖等。

拓扑排序的应用通常使用 BFS(结合入度)来实现。

示例:拓扑排序(使用BFS)

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

class Graph {
public:
    int V;  // 节点数
    vector<vector<int>> adj;  // 邻接表
    vector<int> inDegree;  // 每个节点的入度

    Graph(int V) {
        this->V = V;
        adj.resize(V);
        inDegree.resize(V, 0);
    }

    void addEdge(int u, int v) {
        adj[u].push_back(v);
        inDegree[v]++;  // 目标节点v的入度增加
    }

    vector<int> topologicalSort() {
        queue<int> q;
        vector<int> result;

        // 将所有入度为0的节点入队
        for (int i = 0; i < V; ++i) {
            if (inDegree[i] == 0) {
                q.push(i);
            }
        }

        while (!q.empty()) {
            int node = q.front();
            q.pop();
            result.push_back(node);

            // 遍历所有邻接节点,减少其入度
            for (int neighbor : adj[node]) {
                if (--inDegree[neighbor] == 0) {
                    q.push(neighbor);
                }
            }
        }

        return result;
    }
};

int main() {
    Graph g(6);

    // 添加有向边
    g.addEdge(5, 2);
    g.addEdge(5, 0);
    g.addEdge(4, 0);
    g.addEdge(4, 1);
    g.addEdge(2, 3);
    g.addEdge(3, 1);

    vector<int> result = g.topologicalSort();

    cout << "Topological Sort: ";
    for (int node : result) {
        cout << node << " ";
    }
    cout << endl;

    return 0;
}

总结

  • 深度优先搜索 (DFS):采用递归(或栈)策略,从根节点出发,一直深入到叶子节点。适合在树结构中需要访问每个节点的场景,如路径查找、图的遍历等。
    • 典型实现:前序、中序、后序遍历。
  • 广度优先搜索 (BFS):采用队列策略,从根节点开始,逐层访问节点。适合查找最短路径等场景,尤其在图算法中广泛应用。
    • 典型实现:层序遍历。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值