Proof of correctness: Algorithm for diameter of a tree in graph theory

Question

In order to find the diameter of a tree I can take any node from the tree, perform BFS to find a node which is farthest away from it and then perform BFS on that node. The g

Answer 1

Here's an alternative way to look at it:

Suppose G = ( V, E ) is a nonempty, finite tree with vertex set V and edge set E.

Consider the following algorithm:

1. Let count = 0. Let all edges in E initially be uncolored. Let C initially be equal to V.
2. Consider the subset V' of V containing all vertices with exactly one uncolored edge:
   • if V' is empty then let d = count * 2, and stop.
   • if V' contains exactly two elements then color their mutual (uncolored) edge green, let d = count * 2 + 1, and stop.
   • otherwise, V' contains at least three vertices; proceed as follows:
3. Increment count by one.
4. Remove all vertices from C that have no uncolored edges.
5. For each vertex in V with two or more uncolored edges, re-color each of its green edges red (some vertices may have zero such edges).
6. For each vertex in V', color its uncolored edge green.
7. Return to step (2).

That basically colors the graph from the leaves inward, marking paths with maximal distance to a leaf in green and marking those with only shorter distances in red. Meanwhile, the nodes of C, the center, with shorter maximal distance to a leaf are pared away until C contains only the one or two nodes with the largest maximum distance to a leaf.

By construction, all simple paths from leaf vertices to their nearest center vertex that traverse only green edges are the same length (count), and all other simple paths from a leaf vertex to its nearest center vertex (traversing at least one red edge) are shorter. It can furthermore be proven that

• this algorithm always terminates under the conditions given, leaving every edge of G colored either red or green, and leaving C with either one or two elements.
• at algorithm termination, d is the diameter of G, measured in edges.
• Given a vertex v in V, the maximum-length simple paths in G starting at v are exactly those that contain contain all vertices of the center, terminate at a leaf, and traverse only green edges between center and the far endpoint. These go from v, across the center, to one of the leaves farthest from the center.

---

Now consider your algorithm, which might be more practical, in light of the above. Starting from any vertex v, there is exactly one simple path p from that vertex, ending at a center vertex, and containing all vertices of the center (because G is a tree, and if there are two vertices in C then they share an edge). It can be shown that the maximal simple paths in G having v as one endpoint all have the form of the union of p with a simple path from center to leaf traversing only green edges.

The key point for our purposes is that the incoming edge of the other endpoint is necessarily green. Therefore, when we perform a search for the longest paths starting there, we have access to those traversing only green edges from leaf across (all vertices of) the center to another leaf. Those are exactly the maximal-length simple paths in G, so we can be confident that the second search will indeed reveal the graph diameter.