求二叉树中节点的最大距离

《编程之美》给出了一种方法，但是它要为每个节点都保存一个最长左子树距离和最长右子树距离的字段。

我想到了另一种方法，就是为节点编码，根节点设为0，对于一个节点的左节点，它的编码是它的父亲节点的码后面加个0，对于一个节点的右节点，它的编码是它的父亲节点的码后面加个1，如果我们找到了叶子节点，就把为它编的码存在一个数组中。一次遍历完毕后，数组中就存在所有叶子节点的编码了，再将根节点的码也存在数组中。我们可以根据节点的码来求出两个节点间的距离。因为最大距离的两个节点一定是叶子节点或者叶子节点和根节点，因此只需比较存在数组中的码间的距离，就可以获得最大距离了。

 

汗，怎么每次说思路都说的糊里糊涂的。。。其实，只要画副图就明确了，自己画吧。。。

 

#include <iostream> #include <vector> #include <string> using namespace std; template <class T> struct BTNode { T data; BTNode<T> * left; BTNode<T> * right; BTNode(T i=0,BTNode<T> *l=NULL,BTNode<T> *r=NULL):data(i),left(l),right(r){}; }; vector<string> code; //为叶子节点编码，编号的码存储数组code中。 template <class T> void Code(BTNode<T> *p,string s) { if(p==NULL) return; if(p->left!=NULL) Code(p->left,s+"0"); if(p->right!=NULL) Code(p->right,s+"1"); if(p->left==NULL && p->right==NULL) //叶子节点 code.push_back(s); } template <class T> int maxdistance(BTNode<T> *root) { if(root==NULL) return 0; code.push_back("0"); Code(root,"0"); //编码 int max=0; int i,j; string s1,s2; for(i=0;i<code.size();i++) { for(j=i;j<code.size();j++) { s1=code[i]; s2=code[j]; int k=0; while(k<s1.length() && k<s2.length() && s1.at(k)==s2.at(k)) //根据编的码，计算距离。 k++; int temp=((s1.length()-k)+(s2.length()-k)); if(temp>max) max=temp; } } return max; } int main() { BTNode<char> *h=new BTNode<char>('h'); //注意用模板，注意new BTNode<char> *i=new BTNode<char>('i'); BTNode<char> *d=new BTNode<char>('d',h); BTNode<char> *e=new BTNode<char>('e'); BTNode<char> *b=new BTNode<char>('b',d,e); BTNode<char> *f=new BTNode<char>('f',NULL,i); BTNode<char> *g=new BTNode<char>('g'); BTNode<char> *c=new BTNode<char>('c',f,g); BTNode<char> *a=new BTNode<char>('a',b,c); cout << maxdistance(a) << endl; while(1); return 0; }

 

附上《编程之美》中的另一种解法

 

#include <iostream> using namespace std; #define max(a,b) ((a)>(b)?(a):(b)) struct node { node *left; node *right; int nMaxLeft; //左子树中的最长距离 int nMaxRight; //右子树中的最长距离 char chValue; node(char a,node *l=NULL,node *r=NULL):chValue(a),left(l),right(r){} }; int nMaxLen=0; void FindMaxLen(node *root) { if(root==NULL) return; if(root->left==NULL) root->nMaxLeft=0; if(root->right==NULL) root->nMaxRight=0; if(root->left!=NULL) FindMaxLen(root->left); if(root->right!=NULL) FindMaxLen(root->right); if(root->left!=NULL) root->nMaxLeft=max(root->left->nMaxLeft,root->left->nMaxRight)+1; if(root->right!=NULL) root->nMaxRight=max(root->right->nMaxLeft,root->right->nMaxRight)+1; if(root->nMaxLeft+root->nMaxRight>nMaxLen) nMaxLen=root->nMaxLeft+root->nMaxRight; } int main() { node *h=new node('h'); node *i=new node('i'); node *e=new node('e',h); node *c=new node('c',e); node *f=new node('f',NULL,i); node *g=new node('g'); node *d=new node('d',f,g); node *b=new node('b',c,d); node *a=new node('a',NULL,b); FindMaxLen(a); cout << nMaxLen << endl; while(1); }