经常在图上遇到奥林匹克问题,我总是编写了相当长的代码来实现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,最好采用整洁的方法,该方法采用输入值的形式(开头,结尾)。
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)