DFS / BFS算法的简化版

问题描述 投票:0回答:1

经常在图上遇到奥林匹克问题,我总是编写了相当长的代码来实现DFS,并且调试它花费了很多时间。我写了这样的结构:


class Graph { 
    int V;   
    list<int> *adj;    
public: 
    Graph(int V);  
    void addEdge(int v, int w);  
    void DFS(int s, int f);   

}; 

Graph::Graph(int V) { 
    this->V = V; 
    adj = new list<int>[V]; 
} 

void Graph::addEdge(int v, int w) { 
    adj[v].push_back(w);  
    adj[w].push_back(v);
}

void Graph::DFS(int s,int f) { 
    vector<bool> visited(V, false); 
    stack<int> stack; 
    stack.push(s); 

    while (!stack.empty()) {  
        s = stack.top(); 
        stack.pop(); 

        if (!visited[s]) {
            cout << s << " "; 
            visited[s] = true; 
        } 

        for (list<int>::iterator i = adj[s].begin(); i != adj[s].end(); ++i) 
            if (!visited[*i]) 
                stack.push(*i); 
    }
}

这很长且不便,特别是在将一种语言更改为另一种语言时。从头开始编写时特别困难。我需要一种最佳的方法来尽可能简短,优雅地实现DFS / BFS,最好采用整洁的方法,该方法采用输入值的形式(开头,结尾)。

c++ optimization depth-first-search breadth-first-search
1个回答
0
投票
void addEl(int a, int b,  vector<int> g[]) {
    g[a].push_back(b);
    g[b].push_back(a);
}

void DFS(int s, int f, int n,  vector<int> g[]) {
    vector<bool> vis(n,false);
    stack<int> stk;

    stk.push(s);

    while(!stk.empty()) {
        s = stk.top();
        stk.pop();
        if(!vis[s]) {
            cout << s << endl;
            vis[s] = true;
            if(s==f) return;
        }
        for(int i=0;i<g[s].size();i++) {
            if(!vis[g[s][i]]) stk.push(g[s][i]);
        }
    }
}

可能这就是我想要的(但没有OOP)

© www.soinside.com 2019 - 2024. All rights reserved.