寻找一棵树上最远的两个节点

首先知道这两个节点一定是叶子，或者是根。

 

下面是初步的一些代码，还没完成

 

def makeTree(list): list[0].left=list[1]; list[0].right=list[2]; list[1].left=list[3]; list[1].right=list[4]; list[2].left=list[5]; list[3].left=list[6]; list[3].right=list[7]; list[5].left=list[8]; list[5].right=list[9]; class node:pass; list=[]; for i in range(0,10): n=node() n.data=i; n.left=None; n.right=None; list.append(n); makeTree(list); print list[0].right.data print list[5].right is None def getMaxLength(root): L=getLeftFarthest(root); R=getRightFarthest(root); withRoot=L+R; withoutRoot=bigger(getMaxLength(root.left),getMaxLength(root.right)); return bigger(withRoot,withoutRoot)

 

树结构如下：

 

今天早上把昨天的代码改写了一下，发现找left最远端和找right最远端是一回事。

def bigger(x,y): if x>y: return x; else: return y; def makeTree(list): list[0].left=list[1]; list[0].right=list[2]; list[1].left=list[3]; list[1].right=list[4]; list[2].left=list[5]; list[3].left=list[6]; list[3].right=list[7]; list[5].left=list[8]; list[5].right=list[9]; def getFarthestLeaf(root): if root==None: return 0; if root.left==None: left=0; else: left=getFarthestLeaf(root.left)+1; if root.right==None: right=0; else: right=getFarthestLeaf(root.right)+1; return bigger(left,right); def getMaxLength(root): if root==None: return 0; else: L=getFarthestLeaf(root.left); R=getFarthestLeaf(root.right); withRoot=L+R+2; withoutRoot=bigger(getMaxLength(root.left),getMaxLength(root.right)); return bigger(withRoot,withoutRoot) class node:pass; list=[]; for i in range(0,10): n=node() n.data=i; n.left=None; n.right=None; list.append(n); makeTree(list); print getFarthestLeaf(list[0]); print getMaxLength(list[0]);

 

然后我在想到底是谁和谁之间是最大的距离呢？ 这个问题递归算法里显然没法给出。

我就想把可能的那些点先找出来，就是根节点，或者是叶子节点：

 

def findAllLeaves(root): if root.left==None and root.right==None: leaves.append(root); if root.left!=None: findAllLeaves(root.left); if root.right!=None: findAllLeaves(root.right);

进一步找出这些叶子节点到根的path

def findAllLeaves(root): path_stack.append(root); if root.left==None and root.right==None: leaves.append(root); printDetail(path_stack); if root.left!=None: findAllLeaves(root.left); if root.right!=None: findAllLeaves(root.right); path_stack.remove(root);

 

 

有了这个结构以后，就变成了几个序列焦点，再 找回路的问题：

 

暂时用了两个for去找最小的回路，感觉不好，觉得应该有更好的办法。 继续想想。

 

import copy def makeTree(list): list[0].left=list[1]; list[0].right=list[2]; list[1].left=list[3]; list[1].right=list[4]; list[2].left=list[5]; list[3].left=list[6]; list[3].right=list[7]; list[5].left=list[8]; list[5].right=list[9]; leaves=[]; path_stack=[]; path=[]; def printDetail(list): message='['; for node in list: message=message+str(node.data)+','; message+=']'; print message def findAllLeaves(root): path_stack.append(root); if root.left==None and root.right==None: leaves.append(root); #printDetail(path_stack); #follow code is deepcopy in python path.append(copy.deepcopy(path_stack)); if root.left!=None: findAllLeaves(root.left); if root.right!=None: findAllLeaves(root.right); path_stack.remove(root); def reverse(list):#python 2.6 support list reverse temp=[] for i in range(len(list)-1,-1,-1): temp.append(list[i]); return temp; def genPath(list1,list2): for i in range(len(list1)-1,-1,-1): for j in range(len(list2)-1,-1,-1): if list1[i].data==list2[j].data: listout=reverse(list1[i+1:]); listout.append(list[i]); listout+=list2[j+1:]; return listout def findLongPath(path): longestPath=0; for i in range(0,len(path)): for j in range(i+1,len(path)): realPath=genPath(path[i],path[j]); #printDetail(realPath); if len(realPath)>=longestPath: printDetail(realPath) longestPath=len(realPath) class node:pass; list=[]; for i in range(0,10): n=node() n.data=i; n.left=None; n.right=None; list.append(n); makeTree(list); findAllLeaves(list[0]); leaves.append(list[0]) findLongPath(path);