📜  无向树中的最长路径

📅  最后修改于: 2022-05-13 01:57:17.751000             🧑  作者: Mango

无向树中的最长路径

给定一棵无向树,我们需要找到这棵树的最长路径,其中路径定义为节点序列。

例子:

Input : Below shown Tree using adjacency list 
        representation:
Output : 5
In below tree longest path is of length 5
from node 5 to node 7

这个问题与n叉树的直径相同。我们在这里讨论了一个简单的解决方案。
在这篇文章中,讨论了一个有效的解决方案。我们可以使用两个 BFS 找到最长的路径。这个想法基于以下事实:如果我们从任何节点 x 开始 BFS,并找到一个与 x 距离最长的节点,它必须是最长路径的端点。可以用反证法来证明。所以我们的算法简化为简单的两个 BFS。第一个 BFS 找到最长路径的端点,第二个 BFS 从这个端点找到实际最长的路径。为了证明这个算法为什么有效,这里有一个很好的解释证明正确性:图论中树的直径算法
如上图所示,如果我们从节点 0 开始 BFS,距离它最远的节点将是节点 5,现在如果我们从节点 5 开始 BFS,距离最远的节点将是是node-7,最后,从node-5到node-7的路径将构成我们最长的路径。

C++
// C++ program to find longest path of the tree
#include 
using namespace std;
 
// This class represents a undirected graph using adjacency list
class Graph
{
    int V;              // No. of vertices
    list *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 bfs(int u); // function returns maximum distant
                               // node from u with its distance
};
 
Graph::Graph(int V)
{
    this->V = V;
    adj = new list[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
}
 
//  method returns farthest node and its distance from node u
pair Graph::bfs(int u)
{
    //  mark all distance with -1
    int dis[V];
    memset(dis, -1, sizeof(dis));
 
    queue q;
    q.push(u);
 
    //  distance of u from u will be 0
    dis[u] = 0;
 
    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;
 
            // 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;
            }
        }
    }
 
    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);
}
 
//  method prints longest path of given tree
void Graph::longestPathLength()
{
    pair 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;
}
 
// Driver code to test above methods
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;
}


Java
// Java program to find longest path of the tree
 
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
 
 
class LongestPathUndirectedTree {
     
    // Utility Pair class for storing maximum distance
    // Node with its distance
    static class Pair {
        T first; // maximum distance Node
        V second; // distance of maximum distance node
         
        //Constructor
        Pair(T first, V second) {
            this.first = first;
            this.second = second;
        }
    }
     
    // This class represents a undirected graph using adjacency list
    static class Graph {
        int V; // No. of vertices
        LinkedList[] adj; //Adjacency List
         
        // Constructor
        Graph(int V) {
            this.V = V;
            // Initializing Adjacency List
            adj = new LinkedList[V];
            for(int i = 0; i < V; ++i) {
                adj[i] = new LinkedList();
            }
        }
         
        // function to add an edge to graph
        void addEdge(int s, int d) {
            adj[s].add(d); // Add d to s's list.
            adj[d].add(s); // Since the graph is undirected
        }
         
         
        // method returns farthest node and its distance from node u
        Pair bfs(int u) {
            int[] dis = new int[V];
             
            // mark all distance with -1
            Arrays.fill(dis, -1);
 
            Queue q = new LinkedList<>();
 
            q.add(u);
             
            // distance of u from u will be 0
            dis[u] = 0;
            while (!q.isEmpty()) {
                int t = q.poll();
                 
                // loop for all adjacent nodes of node-t
                for(int i = 0; i < adj[t].size(); ++i) {
                    int v = adj[t].get(i);
                     
                    // push node into queue only if
                    // it is not visited already
                    if(dis[v] == -1) {
                        q.add(v);
                        // make distance of v, one more
                        // than distance of t
                        dis[v] = dis[t] + 1;
                    }
                }
            }
 
            int maxDis = 0;
            int nodeIdx = 0;
             
            // get farthest node distance and its index
            for(int i = 0; i < V; ++i) {
                if(dis[i] > maxDis) {
                    maxDis = dis[i];
                    nodeIdx = i;
                }
            }
 
            return new Pair(nodeIdx, maxDis);
        }
         
        // method prints longest path of given tree
        void longestPathLength() {
            Pair 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);
 
            System.out.println("Longest path is from "+ t1.first
            + " to "+ t2.first +" of length "+t2.second);
        }
    }
     
    // Driver code to test above methods
    public static void main(String[] args){
        // Create a graph given in the example
         
        Graph graph = new Graph(10);
        graph.addEdge(0, 1);
        graph.addEdge(1, 2);
        graph.addEdge(2, 3);
        graph.addEdge(2, 9);
        graph.addEdge(2, 4);
        graph.addEdge(4, 5);
        graph.addEdge(1, 6);
        graph.addEdge(6, 7);
        graph.addEdge(6, 8);
 
        graph.longestPathLength();
    }
 
}
// Added By Brij Raj Kishore


C#
// C# program to find longest path of the tree
using System;
using System.Collections.Generic;
 
class GFG
{
     
// Utility Pair class for storing
// maximum distance Node with its distance
public class Pair
{
    // maximum distance Node
    public T first;
     
    // distance of maximum distance node
    public V second;
     
    // Constructor
    public Pair(T first, V second)
    {
        this.first = first;
        this.second = second;
    }
}
 
// This class represents a undirected graph
// using adjacency list
class Graph
{
    int V; // No. of vertices
    List[] adj; //Adjacency List
     
    // Constructor
    public Graph(int V)
    {
        this.V = V;
         
        // Initializing Adjacency List
        adj = new List[V];
        for(int i = 0; i < V; ++i)
        {
            adj[i] = new List();
        }
    }
     
    // function to add an edge to graph
    public void addEdge(int s, int d)
    {
        adj[s].Add(d); // Add d to s's list.
        adj[d].Add(s); // Since the graph is undirected
    }
     
    // method returns farthest node and
    // its distance from node u
    public Pair bfs(int u)
    {
        int[] dis = new int[V];
         
        // mark all distance with -1
        for(int i = 0; i < V; i++)
            dis[i] = -1;
 
        Queue q = new Queue();
 
        q.Enqueue(u);
         
        // distance of u from u will be 0
        dis[u] = 0;
        while (q.Count != 0)
        {
            int t = q.Dequeue();
             
            // loop for all adjacent nodes of node-t
            for(int i = 0; i < adj[t].Count; ++i)
            {
                int v = adj[t][i];
                 
                // push node into queue only if
                // it is not visited already
                if(dis[v] == -1)
                {
                    q.Enqueue(v);
                     
                    // make distance of v, one more
                    // than distance of t
                    dis[v] = dis[t] + 1;
                }
            }
        }
        int maxDis = 0;
        int nodeIdx = 0;
         
        // get farthest node distance and its index
        for(int i = 0; i < V; ++i)
        {
            if(dis[i] > maxDis)
            {
                maxDis = dis[i];
                nodeIdx = i;
            }
        }
        return new Pair(nodeIdx, maxDis);
    }
     
    // method prints longest path of given tree
    public void longestPathLength()
    {
        Pair 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);
 
        Console.WriteLine("longest path is from " + t1.first +
                " to " + t2.first + " of length " + t2.second);
    }
}
 
// Driver Code
public static void Main(String[] args)
{    
    // Create a graph given in the example
    Graph graph = new Graph(10);
    graph.addEdge(0, 1);
    graph.addEdge(1, 2);
    graph.addEdge(2, 3);
    graph.addEdge(2, 9);
    graph.addEdge(2, 4);
    graph.addEdge(4, 5);
    graph.addEdge(1, 6);
    graph.addEdge(6, 7);
    graph.addEdge(6, 8);
 
    graph.longestPathLength();
}
}
 
// This code is contributed by Rajput-Ji


Python3
# Python program to find the Longest Path of the Tree
# By Aaditya Upadhyay
 
from collections import deque
 
 
class Graph:
 
    # Initialisation of graph
    def __init__(self, vertices):
 
        # No. of vertices
        self.vertices = vertices
 
        # adjacency list
        self.adj = {i: [] for i in range(self.vertices)}
 
    def addEdge(self, u, v):
        # add u to v's list
        self.adj[u].append(v)
        # since the graph is undirected
        self.adj[v].append(u)
 
    # method return farthest node and its distance from node u
    def BFS(self, u):
        # marking all nodes as unvisited
        visited = [False for i in range(self.vertices + 1)]
        # mark all distance with -1
        distance = [-1 for i in range(self.vertices + 1)]
 
        # distance of u from u will be 0
        distance[u] = 0
        # in-built library for queue which performs fast operations on both the ends
        queue = deque()
        queue.append(u)
        # mark node u as visited
        visited[u] = True
 
        while queue:
 
            # pop the front of the queue(0th element)
            front = queue.popleft()
            # loop for all adjacent nodes of node front
 
            for i in self.adj[front]:
                if not visited[i]:
                    # mark the ith node as visited
                    visited[i] = True
                    # make distance of i , one more than distance of front
                    distance[i] = distance[front]+1
                    # Push node into the stack only if it is not visited already
                    queue.append(i)
 
        maxDis = 0
 
        # get farthest node distance and its index
        for i in range(self.vertices):
            if distance[i] > maxDis:
 
                maxDis = distance[i]
                nodeIdx = i
 
        return nodeIdx, maxDis
 
    # method prints longest path of given tree
    def LongestPathLength(self):
 
        # first DFS to find one end point of longest path
        node, Dis = self.BFS(0)
 
        # second DFS to find the actual longest path
        node_2, LongDis = self.BFS(node)
 
        print('Longest path is from', node, 'to', node_2, 'of length', LongDis)
 
 
# create a graph given in the example
 
G = Graph(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()


Javascript


输出:

Longest path is from 5 to 7 of length 5