图的深度优先遍历和广度优先遍历

2 篇文章 0 订阅

不多说,直接上代码

common.h

#pragma once
#include <iostream>
#include <limits.h>
#include <queue>
#include <utility>
#include <vector>

class Graph
{
public:
    int vecnum;
    int edgenum;
    bool *visited;
    std::vector<std::vector<std::pair<int, int>>> vec;
    Graph(const int _vec, const int _edge);
    ~Graph();
    void BFS(int v);
    void DFS(int v);

private:
    void BFSHelper(int u, std::queue<int> &Q);
    void DFSHelper(int u);
    int FirstNeighbor(const int u);
    int NextNeighbor(const int v, const int u);
};

common.cpp

#include "common.h"

Graph::Graph(const int _vec, const int _edge) : vecnum(_vec), edgenum(_edge)
{
    visited = new bool[vecnum]{false};
    for (int i = 0; i < vecnum; i++)
        vec.push_back({});
    int dist, from, to;
    for (int i = 0; i < edgenum; i++)
    {
        std::cin >> from >> to >> dist;
        vec[from].push_back({to, dist});
        vec[to].push_back({from, dist});
    }
    //输出邻接表
    std::cout << std::endl;
    for (int i = 0; i < vecnum; i++)
    {
        std::cout << i << ": ";
        for (int j = 0; j < vec[i].size(); j++)
            std::cout << vec[i][j].first << " ";
        std::cout << std::endl;
    }
}
Graph::~Graph()
{
    delete[] visited;
    std::cout << "\nObj killed." << std::endl;
}
int Graph::FirstNeighbor(const int u)
{
    return !vec[u].size() ? -1 : vec[u][0].first;
}
int Graph::NextNeighbor(const int v, const int u)
{
    int loc = 0;
    for (; vec[v][loc].first != u && loc < vec[v].size(); loc++)
        ;

    return loc + 1 < vec[v].size() ? vec[v][loc + 1].first : -1;
}
void Graph::BFSHelper(int u, std::queue<int> &Q)
{
    std::cout << u << ' ';
    visited[u] = true;
    Q.push(u);
    while (!Q.empty())
    {
        u = Q.front();
        Q.pop();
        for (int i = FirstNeighbor(u); i >= 0; i = NextNeighbor(u, i))
        {
            if (!visited[i])
            {
                std::cout << i << ' ';
                visited[i] = true;
                Q.push(i);
            }
        }
    }
}
void Graph::BFS(int u)
{
    for (int i = 0; i < vecnum; i++)
        visited[i] = false;
    std::queue<int> Q;
    for (int i = u; i < vecnum; i++)
        if (!visited[i])
            BFSHelper(i, Q);
}

void Graph::DFSHelper(int u)
{
    std::cout << u << ' ';
    visited[u] = true;
    for (int i = FirstNeighbor(u); i >= 0; i = NextNeighbor(u, i))
        if (!visited[i])
            DFSHelper(i);
}
void Graph::DFS(int u)
{
    for (int i = 0; i < vecnum; i++)
        visited[i] = false;
    for (int v = u; v < vecnum; v++)
        if (!visited[v])
            DFSHelper(v);
}

main.cpp

#include "common.h"
int main()
{
    int vecnum, edgenum, start;
    std::cout << "Input vecnum: ";
    std::cin >> vecnum;
    std::cout << "Input edgenum: ";
    std::cin >> edgenum;

    Graph G(vecnum, edgenum);

    std::cout << std::endl;
    std::cout << "Where to begin the traverses: ";
    std::cin >> start;

    std::cout << "BFS: ";
    G.BFS(start);
    std::cout << std::endl;

    std::cout << "DFS: ";
    G.DFS(start);
    std::cout << std::endl;

    return 0;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.16.3)
project(GraphTraverse)
add_executable(GraphTraverse main.cpp common.h common.cpp)

几个测试用例

图1在这里插入图片描述

Input vecnum: 8
Input edgenum: 9
0 1 1
0 4 1
0 7 1
1 2 1
1 3 1
2 3 1
4 5 1
4 6 1
2 7 1

0: 1 4 7 
1: 0 2 3 
2: 1 3 7 
3: 1 2 
4: 0 5 6 
5: 4 
6: 4 
7: 0 2 

Where to begin the traverses: 6
BFS: 6 4 0 5 1 7 2 3 
DFS: 6 4 0 1 2 3 7 5 

Obj killed.
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值