读书人

【面试题】在二元树中找到和为某一值的

发布时间: 2012-10-12 10:17:04 作者: rapoo

【面试题】在二元树中找出和为某一值的所有路径
问题:输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。
例如输入整数 22 ,如下图二元树:
10
/ \
5 12
/ \
4 7

则打印出两条路径:10, 12和10, 5, 7。

二元树节点的数据结构定义为:

struct BinaryTreeNode // a node in the binary tree{int m_nValue; // value of nodeBinaryTreeNode *m_pLeft; // left child of nodeBinaryTreeNode *m_pRight; // right child of node};


分析:

思路一

1、当访问到某一节点时,把该结点的值添加到当前和变量,且把该结点压入栈中。

2、若结点为叶子结点,且当前和变量等于期望的和,则打印栈中的结点值,即为所需的路径。

3、若结点不是叶子结点,继续访问它的左孩子结点,访问它的右孩子结点。

4、删除该结点。包括从当前和变量中减去结点值,从栈中弹出结点值。此时,已回到父结点。

思路二

另一种类似的方法不用录当前路径上所有元素的和sum,而是使用期望的和依次减去访问到的节点的值……最后是判断到达叶子节点时期望和是否减为0。

代码如下:

void findPath(BinaryTreeNode *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)     {         vector<int>::iterator iter;         for(iter = path.begin(); iter != path.end(); iter++)         {             cout << *iter << "\t";         }         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(); }int main(){/****创建二叉查找树****/     BinaryTreeNode *pRoot=NULL;     addBSTreeNode(pRoot,10);     addBSTreeNode(pRoot,12);     addBSTreeNode(pRoot,5);     addBSTreeNode(pRoot,7);     addBSTreeNode(pRoot,4);    vector<int> path;    int sum = 0;    findPath(pRoot, 22, path, sum);  return 0;}


读书人网 >编程

热点推荐