题目:
输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。例如输入整数22和如下二元树
10
/ \
5 12
/ \
4 7
则打印出两条路径:10, 12 和10, 5,7。
分析:
当访问到某一结点时,把该结点添加到路径上,并累加当前结点的值。如果当前结点为叶结点并且当前路径的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是根结点到父结点的路径。我们不难看出保存路径的数据结构实际上是一个栈结构,因为路径要与递归调用状态一致,而递归调用本质就是一个压栈和出栈的过程。
- /*Title: 4.在二元树中找出和为某一值的所有路径
- Author: gocode
- Date: 2012-10-08*/
- #include <iostream>
- #include <vector>
- using namespace std;
- // 定义二叉查找树节点
- struct BSTreeNode
- {
- int m_nValue;
- BSTreeNode *m_pLeft;
- BSTreeNode *m_pRight;
- };
- // 查找符合设定和的路径
- // 此处借助vector数组进行压栈、出栈的操作,很方便
- void FindPath(BSTreeNode *pTreeNode, int expectedSum, vector<int>& path, int& currentSum)
- {
- if(!pTreeNode)
- return;
- currentSum += pTreeNode->m_nValue;
- path.push_back(pTreeNode->m_nValue);
- // 如节点是叶子节点,并且相加之和与给定和相同,则打印之
- bool isLeaf = (!pTreeNode->m_pLeft && !pTreeNode->m_pRight);
- if(currentSum == expectedSum && isLeaf)
- {
- for(vector<int>::iterator iter = path.begin(); iter != path.end(); ++ iter)
- cout << *iter << " ";
- cout << endl;
- }
- // 如果节点不是叶子节点,则继续递归调用左、右子树
- if(pTreeNode->m_pLeft)
- FindPath(pTreeNode->m_pLeft, expectedSum, path, currentSum);
- if(pTreeNode->m_pRight)
- FindPath(pTreeNode->m_pRight, expectedSum, path, currentSum);
- // 当访问完一个节点后,返回到其父节点时,从当前和减去,并出栈
- currentSum -= pTreeNode->m_nValue;
- path.pop_back();
- }
- // 添加节点
- // 注意第一个参数,必须加&符号,否则指针不会被改变
- void AddBSTreeNode(BSTreeNode *&pCurrent, int value)
- {
- if(NULL == pCurrent)
- {
- BSTreeNode *pNewNode = new BSTreeNode();
- pNewNode->m_pLeft = NULL;
- pNewNode->m_pRight = NULL;
- pNewNode->m_nValue = value;
- pCurrent = pNewNode;
- }
- else
- {
- if(value < pCurrent->m_nValue)
- AddBSTreeNode(pCurrent->m_pLeft, value);
- else if(value > pCurrent->m_nValue)
- AddBSTreeNode(pCurrent->m_pRight, value);
- else
- cout<<"Duplicate value is not allowed."<<endl;
- }
- }
- void main()
- {
- vector<int> resultpath;
- int resultSum = 0;
- BSTreeNode * pRoot = NULL;
- AddBSTreeNode(pRoot, 10);
- AddBSTreeNode(pRoot, 5);
- AddBSTreeNode(pRoot, 12);
- AddBSTreeNode(pRoot, 4);
- AddBSTreeNode(pRoot, 7);
- cout<<"Result is: "<<endl;
- FindPath(pRoot, 22, resultpath, resultSum);
- system("paus