圖的BFS和DFS之C++實現

圖的創建

  1. 手動輸入,並以節點數作爲輸入結束標誌
  2. 從文件讀取,文件的從第二行開始,每一行結束都要有節點數作爲結束
    圖的存儲:
vector< list<int> > graph;

圖的BFS

非遞歸實現,藉助隊列

void bfs(int v)//以v開始做廣度優先搜索(非遞歸實現,藉助隊列)
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    queue<int> myque;
    myque.push(v);
    while (!myque.empty())
    {
        v = myque.front();
        myque.pop();
        for (it = graph[v].begin(); it != graph[v].end(); it++)
        {
            if (!visited[*it])
            {
                cout << *it << " ";
                myque.push(*it);
                visited[*it] = true;//訪問過
            }
        }
    }
    cout << endl;
}

圖的DFS

遞歸和非遞歸

void dfs(int v)//以v開始做深度優先搜索
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    for (it = graph[v].begin(); it != graph[v].end(); it++)
        if (!visited[*it])
            dfs(*it);
}
void dfs_noRecursion(int v)//以v開始做深度優先搜索,非遞歸實現
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    stack<int>mystack;
    mystack.push(v);
    while (!mystack.empty())
    {
        v = mystack.top();
        mystack.pop();
        if (!visited[v])
        {
            cout << v << " ";
            visited[v] = true;
        }

        for (it = graph[v].begin(); it != graph[v].end(); it++)
        {
            if (!visited[*it])
            {
                mystack.push(*it);

            }
        }
    }
    cout << endl;
}

測試

#include<iostream>
#include <cstdio>
#include <list>
#include <vector>
#include <queue>
#include<stack>
#include<fstream>
using namespace std;
int n;
vector< list<int> > graph;
bool visited[100] = { 0 };
void CreateGraph()
{
    cout << "input the vertex num:" << endl;
    cin >> n;
    vector< list<int> >::iterator it;
    for (int i = 0; i < n; ++i)
    {
        list<int> i1;
        int t;
        while (cin >> t&&t != n)//輸入每個頂點相連的圖,以最終節點爲結束標誌
            i1.push_back(t);
        graph.push_back(i1);
    }
}
void CreateGraph1()
{
    int n;
    vector< list<int> >::iterator it;
    ifstream in("data.txt");
    in >> n;
    for (int i = 0; i < n; ++i)
    {
        list<int> i1;
        int t;
        while (in >> t&&t != n)//輸入每個頂點相連的圖,以最終節點爲結束標誌
            i1.push_back(t);
        graph.push_back(i1);
    }
}
void dfs(int v)//以v開始做深度優先搜索
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    for (it = graph[v].begin(); it != graph[v].end(); it++)
        if (!visited[*it])
            dfs(*it);
}
void dfs_noRecursion(int v)//以v開始做深度優先搜索,非遞歸實現
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    stack<int>mystack;
    mystack.push(v);
    while (!mystack.empty())
    {
        v = mystack.top();
        mystack.pop();
        if (!visited[v])
        {
            cout << v << " ";
            visited[v] = true;
        }

        for (it = graph[v].begin(); it != graph[v].end(); it++)
        {
            if (!visited[*it])
            {
                mystack.push(*it);

            }
        }
    }
    cout << endl;
}
void bfs(int v)//以v開始做廣度優先搜索(非遞歸實現,藉助隊列)
{
    list<int>::iterator it;
    visited[v] = true;
    cout << v << " ";
    queue<int> myque;
    myque.push(v);
    while (!myque.empty())
    {
        v = myque.front();
        myque.pop();
        for (it = graph[v].begin(); it != graph[v].end(); it++)
        {
            if (!visited[*it])
            {
                cout << *it << " ";
                myque.push(*it);
                visited[*it] = true;//訪問過
            }
        }
    }
    cout << endl;
}
int main()
{
    CreateGraph1();
    cout << "result for bfs:" << endl;
    bfs(0);
    memset(visited, 0, sizeof(visited));
    cout << "result for dfs(Recursion):" << endl;
    dfs(0);
    cout << endl;
    memset(visited, 0, sizeof(visited));
    cout << "result for nodfs(Recursion):" << endl;
    dfs_noRecursion(0);
    system("pause");
    return 0;
}

參考:

http://www.cnblogs.com/PegasusWang/archive/2013/04/06/3002511.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章