0
我需要计算树中两个节点之间的距离。 我实现了通式:树中两个节点之间的距离算法问题
Dist(n1,n2) = Dist(root,n1), Dist(root,n2) - 2*Dist(root,lca)
代码适用于大多数的投入,但如果我尝试用(父亲,儿子)我得到错误的距离。 这里是我的代码:
public static int distance(Tree t, Node<String> node1, Node<String> node2)
{
Node<String> lca = lca(node1, node2);
//This if statement is when the input contains the root itself
//this means that the distance is simply from the node and the root
if(lca==null)
{
// the function getNumberOfAncestors returns exactly the distance between a generic node and the root, since I count parent by parent and so on.
return node1.getNumberOfAncestors() + node2.getNumberOfAncestors();
}
return node1.getNumberOfAncestors() + node2.getNumberOfAncestors() - 2*distance(t, t.getRoot(), lca);
}
对于简单的树:
A
B C
D E F G
我们知道,DIST(C,G)= 1,但我的算法返回3 任何想法?
试试BFS。它非常简单,记录完备,并且还会为您提供启动节点和其他任何节点之间的最短路径。 – beatngu13
对于调试,首先应打印node1.getNumberOfAncestors(),node2.getNumberOfAncestors()和distance(t,t.getRoot(),lca)。查看所有3个值是否正确。编辑:也有一个原因,你不使用lca的getNumberOfAncestors()? –