《数据结构》C++代码 BFS与DFS

#include <iostream>
#include <cstdio>
using namespace std;

const int maxn = 10000;

int N,M,Start,End;
struct edge
{
    int v;
    edge *next;
    
    edge(int _v=0) { v=_v; }
}*E[maxn];
void AddEdge(int u,int v)
{
    edge *p=new edge(v);
    p->next=E[u]; E[u]=p;
}

void input()
{
    cin>>N>>M>>Start>>End;
    
    for(int i=1;i<=N;++i) E[i]=0;
    
    int u,v;
    for(int i=1;i<=M;++i)
    {
        cin>>u>>v;
        AddEdge(u,v);
    }
}

bool mark[maxn];
void DFS(int u)
{
    if(mark[u]) return;

    cout<<u<<" "; mark[u]=true;
    for(edge *p=E[u];p;p=p->next)
    {
        int v=p->v;
        DFS(v);
    }
}

class Queue
{
    int Q[maxn],i,j;
public:
    Queue() { i=j=0; }
    void clear() { i=j=0; }
    void in(int u) { Q[j++]=u; }
    int out() { return Q[i++]; }
    bool empty() { return i==j; }
    int size() { return j-i; }
};

Queue Q;
void BFS(int u)
{
    Q.clear();
    Q.in(u); mark[u]=true;
    while(!Q.empty())
    {
        u=Q.out();
        cout<<u<<" ";
        for(edge *p=E[u];p;p=p->next)
        {
            int v=p->v;
            if(mark[v]==false) { Q.in(v); mark[v]=true; }
        }
    }
}

int main()
{
    freopen("map1.in","r",stdin);
    
    input();
    
    for(int i=1;i<=N;++i) mark[i]=false;
    for(int i=1;i<=N;++i) DFS(i);
    cout<<endl;
    
    for(int i=1;i<=N;++i) mark[i]=false;
    for(int i=1;i<=N;++i)
        if(mark[i]==false) BFS(i);
    cout<<endl;
    
    fclose(stdin);
    return 0;
}