第十周项目3——利用二叉树遍历思想解决问题

1. /*   
2. * Copyright (c)2016,烟台大学计算机与控制工程学院   
3. * All rights reserved.   
4. * 文件名称：wu.cpp   
5. * 作    者：武昊   
6. * 完成日期：2016年11月3日   
7. * 版 本 号：v1.0    
8. *问题描述：假设二叉树采用二叉链存储结构存储，分别实现以下算法，并在程序中完成测试：  
9. 　　（1）计算二叉树节点个数；  
10. 　　（2）输出所有叶子节点；  
11. 　　（3）求二叉树b的叶子节点个数；  
12. 　　（4）设计一个算法Level(b,x,h)，返回二叉链b中data值为x的节点的层数。  
13. 　　（5）判断二叉树是否相似（关于二叉树t1和t2相似的判断：①t1和t2都是空的二叉树，相似； 
14.     ②t1和t2之一为空，另一不为空，则不相似；③t1的左子树和t2的左子树是相似的，且t1的右子树与t2的右子树是相似的，则t1和t2相似。）  
15. *输入描述：无   
16. *程序输出：测试数据   
17. */

btree.h：

1. #include <stdio.h>  
2. #include <malloc.h>  
3. #define MaxSize 100  
4. typedef char ElemType;  
5. typedef struct node  
6. {  
7.     ElemType data;              //数据元素  
8.     struct node *lchild;        //指向左孩子  
9.     struct node *rchild;        //指向右孩子  
10. } BTNode;  
11. void CreateBTNode(BTNode *&b,char *str);        //由str串创建二叉链  
12. BTNode *FindNode(BTNode *b,ElemType x);     //返回data域为x的节点指针  
13. BTNode *LchildNode(BTNode *p);  //返回*p节点的左孩子节点指针  
14. BTNode *RchildNode(BTNode *p);  //返回*p节点的右孩子节点指针  
15. int BTNodeDepth(BTNode *b); //求二叉树b的深度  
16. void DispBTNode(BTNode *b); //以括号表示法输出二叉树  
17. void DestroyBTNode(BTNode *&b);  //销毁二叉树

btree.cpp：

1. #include "btree.h"  
2.   
3. void CreateBTNode(BTNode *&b,char *str)     //由str串创建二叉链  
4. {  
5.     BTNode *St[MaxSize],*p=NULL;  
6.     int top=-1,k,j=0;  
7.     char ch;  
8.     b=NULL;             //建立的二叉树初始时为空  
9.     ch=str[j];  
10.     while (ch!='\0')    //str未扫描完时循环  
11.     {  
12.         switch(ch)  
13.         {  
14.         case '(':  
15.             top++;  
16.             St[top]=p;  
17.             k=1;  
18.             break;      //为左节点  
19.         case ')':  
20.             top--;  
21.             break;  
22.         case ',':  
23.             k=2;  
24.             break;                          //为右节点  
25.         default:  
26.             p=(BTNode *)malloc(sizeof(BTNode));  
27.             p->data=ch;  
28.             p->lchild=p->rchild=NULL;  
29.             if (b==NULL)                    //p指向二叉树的根节点  
30.                 b=p;  
31.             else                            //已建立二叉树根节点  
32.             {  
33.                 switch(k)  
34.                 {  
35.                 case 1:  
36.                     St[top]->lchild=p;  
37.                     break;  
38.                 case 2:  
39.                     St[top]->rchild=p;  
40.                     break;  
41.                 }  
42.             }  
43.         }  
44.         j++;  
45.         ch=str[j];  
46.     }  
47. }  
48. BTNode *FindNode(BTNode *b,ElemType x)  //返回data域为x的节点指针  
49. {  
50.     BTNode *p;  
51.     if (b==NULL)  
52.         return NULL;  
53.     else if (b->data==x)  
54.         return b;  
55.     else  
56.     {  
57.         p=FindNode(b->lchild,x);  
58.         if (p!=NULL)  
59.             return p;  
60.         else  
61.             return FindNode(b->rchild,x);  
62.     }  
63. }  
64. BTNode *LchildNode(BTNode *p)   //返回*p节点的左孩子节点指针  
65. {  
66.     return p->lchild;  
67. }  
68. BTNode *RchildNode(BTNode *p)   //返回*p节点的右孩子节点指针  
69. {  
70.     return p->rchild;  
71. }  
72. int BTNodeDepth(BTNode *b)  //求二叉树b的深度  
73. {  
74.     int lchilddep,rchilddep;  
75.     if (b==NULL)  
76.         return(0);                          //空树的高度为0  
77.     else  
78.     {  
79.         lchilddep=BTNodeDepth(b->lchild);   //求左子树的高度为lchilddep  
80.         rchilddep=BTNodeDepth(b->rchild);   //求右子树的高度为rchilddep  
81.         return (lchilddep>rchilddep)? (lchilddep+1):(rchilddep+1);  
82.     }  
83. }  
84. void DispBTNode(BTNode *b)  //以括号表示法输出二叉树  
85. {  
86.     if (b!=NULL)  
87.     {  
88.         printf("%c",b->data);  
89.         if (b->lchild!=NULL || b->rchild!=NULL)  
90.         {  
91.             printf("(");  
92.             DispBTNode(b->lchild);  
93.             if (b->rchild!=NULL) printf(",");  
94.             DispBTNode(b->rchild);  
95.             printf(")");  
96.         }  
97.     }  
98. }  
99. void DestroyBTNode(BTNode *&b)   //销毁二叉树  
100. {  
101.     if (b!=NULL)  
102.     {  
103.         DestroyBTNode(b->lchild);  
104.         DestroyBTNode(b->rchild);  
105.         free(b);  
106.     }  
107. }  

main.cpp：

1. #include "btree.h"  
2.   
3. //（1）计算二叉树节点个数；  
4. int Nodes(BTNode *b)  
5. {  
6.     if (b==NULL)  
7.         return 0;  
8.     else  
9.         return Nodes(b->lchild)+Nodes(b->rchild)+1;  
10. }  
11. //（2）输出所有叶子节点；  
12. void DispLeaf(BTNode *b)  
13. {  
14.     if (b!=NULL)  
15.     {  
16.         if (b->lchild==NULL && b->rchild==NULL)  
17.             printf("%c ",b->data);  
18.         else  
19.         {  
20.             DispLeaf(b->lchild);  
21.             DispLeaf(b->rchild);  
22.         }  
23.     }  
24. }  
25. //（3）求二叉树b的叶子节点个数  
26. int LeafNodes(BTNode *b)    //求二叉树b的叶子节点个数  
27. {  
28.     int num1,num2;  
29.     if (b==NULL)  
30.         return 0;  
31.     else if (b->lchild==NULL && b->rchild==NULL)  
32.         return 1;  
33.     else  
34.     {  
35.         num1=LeafNodes(b->lchild);  
36.         num2=LeafNodes(b->rchild);  
37.         return (num1+num2);  
38.     }  
39. }  
40. //（4）设计一个算法Level(b,x,h)，返回二叉链b中data值为x的节点的层数。  
41. int Level(BTNode *b,ElemType x,int h)  
42. {  
43.     int l;  
44.     if (b==NULL)  
45.         return 0;  
46.     else if (b->data==x)  
47.         return h;  
48.     else  
49.     {  
50.         l=Level(b->lchild,x,h+1);  
51.         if (l==0)  
52.             return Level(b->rchild,x,h+1);  
53.         else  
54.             return l;  
55.     }  
56. }  
57. //（5）判断二叉树是否相似（关于二叉树t1和t2相似的判断：  
58. //   ①t1和t2都是空的二叉树，相似；  
59. //   ②t1和t2之一为空，另一不为空，则不相似；  
60. //   ③t1的左子树和t2的左子树是相似的，且t1的右子树与t2的右子树是相似的，则t1和t2相似。）  
61. int Like(BTNode *b1,BTNode *b2)  
62. {  
63.     int like1,like2;  
64.     if (b1==NULL && b2==NULL)  
65.         return 1;  
66.     else if (b1==NULL || b2==NULL)  
67.         return 0;  
68.     else  
69.     {  
70.         like1=Like(b1->lchild,b2->lchild);  
71.         like2=Like(b1->rchild,b2->rchild);  
72.         return (like1 & like2);  
73.     }  
74. }  
75.   
76. int main()  
77. {  
78.     BTNode *b;  
79.     CreateBTNode(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))");  
80.     printf("（1）二叉树节点个数: %d\n\n", Nodes(b));  
81.   
82.     printf("（2）二叉树中所有的叶子节点是: ");  
83.     DispLeaf(b);printf("\n\n");  
84.   
85.     printf("（3）二叉树b的叶子节点个数: %d\n\n",LeafNodes(b));  
86.   
87.     printf("（4）值为\'K\'的节点在二叉树中出现在第 %d 层上n\n\n",Level(b,'K',1));  
88.   
89.     BTNode *b1, *b2, *b3;  
90.     CreateBTNode(b1,"B(D,E(H(J,K(L,M(,N)))))");  
91.     CreateBTNode(b2,"A(B(D(,G)),C(E,F))");  
92.     CreateBTNode(b3,"u(v(w(,x)),y(z,p))");  
93.     if(Like(b1, b2))  
94.         printf("（5）b1和b2相似\n");  
95.     else  
96.         printf("（5）b1和b2不相似\n");  
97.     if(Like(b2, b3))  
98.         printf("     b2和b3相似\n");  
99.     else  
100.         printf("     b2和b3不相似\n");  
101.     DestroyBTNode(b1);  
102.     DestroyBTNode(b2);  
103.     DestroyBTNode(b3);  
104.   
105.     DestroyBTNode(b);  
106.     return 0;  
107. }  

运行结果：