我正在使用此代码https://www.geeksforgeeks.org/longest-path-undirected-tree/来查找无向图中的最长路径。该代码使用两次 BFS 搜索来找到最长的路径,然后输出路径的开始和结束以及长度。如何将路径保存在列表中并打印出来?我将前辈保存在一个数组int predecessors[n]
中,但这当然没有给出路径。我知道我应该以某种方式修改它,pred[V]
以便它存储前辈列表,但我不知道如何实现它。任何帮助表示赞赏。
// C++ program to find longest path of the tree
#include <bits/stdc++.h>
using namespace std;
// This class represents a undirected graph using adjacency list
class Graph {
int V; // No. of vertices
list<int> *adj; // Pointer to an array containing adjacency lists
public:
Graph(int V); // Constructor
void addEdge(int v, int w);// function to add an edge to graph
void longestPathLength(); // prints longest path of the tree
pair<int, int> bfs(int u); // function returns maximum distant
// node from u with its distance
};
Graph::Graph(int V)
{
this->V = V;
adj = new list<int>[V];
}
void Graph::addEdge(int v, int w)
{
adj[v].push_back(w); // Add w to v’s list.
adj[w].push_back(v); // Since the graph is undirected
}
// 方法返回最远节点及其到节点 u 的距离
pair<int, int> Graph::bfs(int u)
{
// mark all distance with -1
int dis[V];
int pred[V]; \\ I added this to store predecessors
memset(dis, -1, sizeof(dis));
queue<int> q;
q.push(u);
dis[u] = 0; // distance of u from u will be 0
pred[u] = {u}; // I added this line
while (!q.empty())
{
int t = q.front(); q.pop();
// loop for all adjacent nodes of node-t
for (auto it = adj[t].begin(); it != adj[t].end(); it++)
{
int v = *it;
cout << "adjacent node:" << v << endl;
// push node into queue only if it is not visited already
if (dis[v] == -1)
{
q.push(v);
// make distance of v, one more than distance of t
dis[v] = dis[t] + 1;
cout << "parent of adjacent node:" << t << endl;
pred[v] = t // store the predecessor of v
}
}
}
int maxDis = 0;
int nodeIdx;
// get farthest node distance and its index
for (int i = 0; i < V; i++)
{
if (dis[i] > maxDis)
{
maxDis = dis[i];
nodeIdx = i;
}
}
return make_pair(nodeIdx, maxDis);
}
// 方法打印给定树的最长路径
void Graph::longestPathLength()
{
pair<int, int> t1, t2;
// first bfs to find one end point of longest path
t1 = bfs(0);
// second bfs to find actual longest path
t2 = bfs(t1.first);
cout << "Longest path is from " << t1.first << " to "
<< t2.first << " of length " << t2.second;
}
// 测试上述方法的驱动代码
int main()
{
// Create a graph given in the example
Graph g(10);
g.addEdge(0, 1);
g.addEdge(1, 2);
g.addEdge(2, 3);
g.addEdge(2, 9);
g.addEdge(2, 4);
g.addEdge(4, 5);
g.addEdge(1, 6);
g.addEdge(6, 7);
g.addEdge(6, 8);
g.longestPathLength();
return 0;
}
// 结果:
Longest path is from 5 to 7 of length 5