在二元树中找出和为某一值的所有路径

具体实现代码如下:

BSTree.h具体内容:

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. #ifndef _BSTREE_H_  
  2. #define _BSTREE_H_  
  3. typedef struct _tagBSTreeNode  
  4. {  
  5.     int m_nVal;  
  6.     _tagBSTreeNode * m_pLeft;  
  7.     _tagBSTreeNode * m_pRight;  
  8. }BSTreeNode;  
  9. extern BSTreeNode * pHead;  
  10. //追加树结点  
  11. int AppendNode(BSTreeNode **, int nNum);  
  12. //打印树结点  
  13. void PrintTree(BSTreeNode ** ppTree);  
  14. //释放树  
  15. void Clear(BSTreeNode ** ppTree);  
  16. //转换为有序双链表  
  17. void InOrderTree(BSTreeNode ** ppTree);  
  18. //销毁链表  
  19. void ClearList(BSTreeNode * pHead);  
  20. //打印链表  
  21. void PrintList(BSTreeNode * pHead);  
  22. //打印路径  
  23. void PrintPath(BSTreeNode * pHead, int nSum);  
  24. //打印和路径  
  25. void PrintSumPath(BSTreeNode ** ppTree, int nLayer, int nPathSum);  
  26. #endif  

BSTree.cpp的内容:

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. #include <stdlib.h>  
  2. #include <stdio.h>  
  3. #include "BSTree.h"  
  4.   
  5. //追加树结点  
  6. int AppendNode(BSTreeNode ** ppTree, int nNum)  
  7. {  
  8.     BSTreeNode * pNewNode = NULL;  
  9.     if (!ppTree)  
  10.         return 0;  
  11.     if (*ppTree == NULL)  
  12.     {  
  13.         pNewNode = (BSTreeNode *)malloc(sizeof(BSTreeNode));  
  14.         if (!pNewNode)  
  15.             return 0;  
  16.         pNewNode->m_nVal = nNum;  
  17.         pNewNode->m_pLeft = NULL;  
  18.         pNewNode->m_pRight = NULL;  
  19.         *ppTree = pNewNode;  
  20.         return 1;  
  21.     }  
  22.     if ((*ppTree)->m_nVal > nNum)  
  23.     {  
  24.         AppendNode(&((*ppTree)->m_pLeft), nNum);  
  25.     }  
  26.     else if ((*ppTree)->m_nVal < nNum)  
  27.     {  
  28.         AppendNode(&((*ppTree)->m_pRight), nNum);  
  29.     }  
  30.     else  
  31.     {  
  32.         printf("树结点已经存在.\n");  
  33.         return 0;  
  34.     }  
  35.     return 1;  
  36. }  
  37. //打印树结点  
  38. void PrintTree(BSTreeNode ** ppTree)  
  39. {  
  40.     if (!ppTree)  
  41.         return;  
  42.     if (*ppTree == NULL)  
  43.     {  
  44.         return;  
  45.     }  
  46.     if ((*ppTree)->m_pLeft)  
  47.         PrintTree(&((*ppTree)->m_pLeft));  
  48.     if (*ppTree)  
  49.         printf("%d", (*ppTree)->m_nVal);  
  50.     if ((*ppTree)->m_pRight)  
  51.         PrintTree(&((*ppTree)->m_pRight));  
  52. }  
  53.   
  54. //释放树  
  55. void Clear(BSTreeNode ** ppTree)  
  56. {  
  57.     BSTreeNode * pLeftTree = NULL;  
  58.     BSTreeNode * pRightTree = NULL;  
  59.     if (!ppTree)  
  60.         return;  
  61.     if (*ppTree == NULL)  
  62.         return;  
  63.     if ((*ppTree)->m_pLeft)  
  64.     {  
  65.         pLeftTree = (*ppTree)->m_pLeft;  
  66.         Clear(&pLeftTree);  
  67.     }  
  68.     if ((*ppTree)->m_pRight)  
  69.     {  
  70.         pRightTree = (*ppTree)->m_pRight;  
  71.         Clear(&pRightTree);  
  72.     }  
  73.     if (*ppTree)  
  74.     {  
  75.         free(*ppTree);  
  76.         *ppTree = NULL;  
  77.     }  
  78.     return;  
  79. }  
  80.   
  81. //以左孩子为前驱  
  82. //以右孩子为后继  
  83. BSTreeNode * pHead = NULL;  
  84.   
  85. void AppendNode2List(BSTreeNode * pInput)  
  86. {  
  87.     BSTreeNode * pCurNode = NULL;  
  88.     if (pInput == NULL)  
  89.         return;  
  90.     if (!pHead)  
  91.     {  
  92.         pHead = pInput;  
  93.         pInput->m_pLeft = NULL;  
  94.         pInput->m_pRight = NULL;  
  95.         return;  
  96.     }  
  97.     pCurNode = pHead;  
  98.     while (pCurNode->m_pRight)  
  99.     {  
  100.         pCurNode = pCurNode->m_pRight;  
  101.     }  
  102.     pCurNode->m_pRight = pInput;  
  103.     pInput->m_pLeft = pCurNode;  
  104.     pInput->m_pRight = NULL;  
  105.     return ;  
  106. }  
  107.   
  108. //打印树结点  
  109. void InOrderTree(BSTreeNode ** ppTree)  
  110. {  
  111.     if (!ppTree)  
  112.         return;  
  113.     if (*ppTree == NULL)  
  114.     {  
  115.         return;  
  116.     }  
  117.     if ((*ppTree)->m_pLeft)  
  118.         InOrderTree(&((*ppTree)->m_pLeft));  
  119.     if (*ppTree)  
  120.         AppendNode2List(*ppTree);  
  121.     if ((*ppTree)->m_pRight)  
  122.         InOrderTree(&((*ppTree)->m_pRight));  
  123. }  
  124.   
  125. void PrintList(BSTreeNode * pHead)  
  126. {  
  127.     BSTreeNode * pCurNode = NULL;  
  128.     if (!pHead)  
  129.         return;  
  130.     pCurNode = pHead;  
  131.     printf("\n链表数据:\n");  
  132.     while (pCurNode)  
  133.     {  
  134.         printf("%d  ", pCurNode->m_nVal);  
  135.         pCurNode = pCurNode->m_pRight;  
  136.     }  
  137.     return;  
  138. }  
  139.   
  140. void ClearList(BSTreeNode * pHead)  
  141. {  
  142.     BSTreeNode * pCurNode = NULL;  
  143.     if (!pHead)  
  144.         return;  
  145.     while (pHead)  
  146.     {  
  147.         pCurNode = pHead->m_pRight;  
  148.         free(pHead);  
  149.         pHead = pCurNode;  
  150.     }  
  151.     return;  
  152. }  
  153.   
  154. //打印和路径  
  155. void PrintSumPath(BSTreeNode ** ppTree, int nLayer, int nPathSum)  
  156. {  
  157.     static int Sum[256] = { 0 };  
  158.     int nSum = 0;  
  159.     int i = 0;  
  160.     if (!ppTree)  
  161.         return;  
  162.     if (*ppTree == NULL)  
  163.     {  
  164.         return;  
  165.     }  
  166.     if (*ppTree)  
  167.     {  
  168.         Sum[nLayer] = (*ppTree)->m_nVal;  
  169.     }     
  170.     if ((*ppTree)->m_pLeft)  
  171.         PrintSumPath(&((*ppTree)->m_pLeft), nLayer + 1, nPathSum);  
  172.     if ((*ppTree)->m_pRight)  
  173.         PrintSumPath(&((*ppTree)->m_pRight), nLayer + 1, nPathSum);  
  174.     if (((*ppTree)->m_pLeft==NULL) && ((*ppTree)->m_pRight == NULL))  
  175.     {  
  176.         for (i = 0; i <= nLayer; i++)  
  177.         {  
  178.             nSum += Sum[i];           
  179.         }  
  180.         if (nSum == nPathSum)  
  181.         {  
  182.             printf("路径:\n");  
  183.             for (i = 0; i <= nLayer; i++)  
  184.             {  
  185.                 printf("%d  ", Sum[i]);  
  186.             }  
  187.             printf("\n");  
  188.         }  
  189.     }  
  190.     return;  
  191. }  
main.cpp的内容:

[cpp]  view plain copy 在CODE上查看代码片 派生到我的代码片
  1. #define _CRTDBG_MAP_ALLOC  
  2. #include <stdlib.h>  
  3. #include <stdio.h>  
  4. #include <crtdbg.h>  
  5. #include "BSTree.h"  
  6.   
  7. void main()  
  8. {  
  9.     BSTreeNode * pTree = NULL;  
  10.     if (AppendNode(&pTree, 10) == NULL)  
  11.     {  
  12.         printf("追加失败.\n");  
  13.         return;  
  14.     }  
  15.     if (AppendNode(&pTree, 5) == NULL)  
  16.     {  
  17.         printf("追加失败.\n");  
  18.         return;  
  19.     }  
  20.     if (AppendNode(&pTree, 12) == NULL)  
  21.     {  
  22.         printf("追加失败.\n");  
  23.         return;  
  24.     }  
  25.     if (AppendNode(&pTree, 4) == NULL)  
  26.     {  
  27.         printf("追加失败.\n");  
  28.         return;  
  29.     }  
  30.     if (AppendNode(&pTree, 7) == NULL)  
  31.     {  
  32.         printf("追加失败.\n");  
  33.         return;  
  34.     }  
  35.     PrintTree(&pTree);  
  36.     PrintSumPath(&pTree, 0,22); //找出和为22的值  
  37.     Clear(&pTree);  
  38.     _CrtDumpMemoryLeaks();  
  39.     system("pause");  
  40.     return;  
  41. }  
运行效果如图1所示:

图1 运行效果

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值