题目:输入一个整数和一棵二元树。
从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。
打印出和与输入整数相等的所有路径。
例如 输入整数22和如下二元树
10
/ \
5 12
/ \
4 7
则打印出两条路径:10, 12和10, 5, 7。
采用先序遍历以及 栈的特性 后进先出:
void PrintPath(BTree* pRoot,int sum, const int key) { static deque<int> stack; if (pRoot == NULL) { return; } if (sum+pRoot->m_nValue == key) //如果当前值加上路径和为目标值,则输出 { for (int i = 0; i < stack.size(); i++) { cout<<stack[i]<<"->"; } cout<<pRoot->m_nValue<<endl; return; } else if (sum + pRoot->m_nValue > key)//如果大于目标值,则返回 { return; } else//如果小于,则入栈 { stack.push_back(pRoot->m_nValue); sum += pRoot->m_nValue; PrintPath(pRoot->m_nLeft,sum,key); PrintPath(pRoot->m_nRight,sum,key); sum -= pRoot->m_nValue; stack.pop_back(); } }
完整测试代码:
// PrintPath.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <iostream> #include <list> #include <queue> #include <deque> using namespace std; //节点的数据结构 class BTree { public: int m_nValue; BTree* m_nLeft; BTree* m_nRight; public: BTree(int value) { m_nValue = value; } }; //二叉树的插入实现 void Insert(int value, BTree* &root) { if (root == NULL) { root = new BTree(value); } else if(value < root->m_nValue) Insert(value,root->m_nLeft); else if(value > root->m_nValue) Insert(value,root->m_nRight); else ; } void PrintPath(BTree* pRoot,int sum, const int key) { static deque<int> stack; if (pRoot == NULL) { return; } if (sum+pRoot->m_nValue == key) //如果当前值加上路径和为目标值,则输出 { for (int i = 0; i < stack.size(); i++) { cout<<stack[i]<<"->"; } cout<<pRoot->m_nValue<<endl; return; } else if (sum + pRoot->m_nValue > key)//如果大于目标值,则返回 { return; } else//如果小于,则入栈 { stack.push_back(pRoot->m_nValue); sum += pRoot->m_nValue; PrintPath(pRoot->m_nLeft,sum,key); PrintPath(pRoot->m_nRight,sum,key); sum -= pRoot->m_nValue; stack.pop_back(); } } int _tmain(int argc, _TCHAR* argv[]) { BTree* m_pRoot = new BTree(5); Insert(7,m_pRoot); Insert(3,m_pRoot); Insert(4,m_pRoot); Insert(2,m_pRoot); PrintPath(m_pRoot,0,12); getchar(); return 0; }
时间: 2024-10-11 22:33:39