二元树中和为某一值的所有路径
题目:输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。
例如输入整数22和如下二元树
10
/ \
5 12
/ \
4 7
则打印出两条路径:10, 12和10, 5, 7。
二元树结点的数据结构定义为:
1 struct BinaryTreeNode // a node in the binary tree 2 { 3 int m_nValue; // value of node 4 BinaryTreeNode *m_pLeft; // left child of node 5 BinaryTreeNode *m_pRight; // right child of node 6 };
分析:这是百度的一道笔试题,考查对树这种基本数据结构以及递归函数的理解。
当访问到某一结点时,把该结点添加到路径上,并累加当前结点的值。如果当前结点为叶结点并且当前路径的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是根结点到父结点的路径。我们不难看出保存路径的数据结构实际上是一个栈结构,因为路径要与递归调用状态一致,而递归调用本质就是一个压栈和出栈的过程。
参考代码:
1 /////////////////////////////////////////////////////////////////////// 2 // Find paths whose sum equal to expected sum 3 /////////////////////////////////////////////////////////////////////// 4 void FindPath 5 ( 6 BinaryTreeNode* pTreeNode, // a node of binary tree 7 int expectedSum, // the expected sum 8 std::vector<int>& path, // a path from root to current node 9 int& currentSum // the sum of path 10 ) 11 { 12 if(!pTreeNode) 13 return; 14 15 currentSum += pTreeNode->m_nValue; 16 path.push_back(pTreeNode->m_nValue); 17 18 // if the node is a leaf, and the sum is same as pre-defined, 19 // the path is what we want. print the path 20 bool isLeaf = (!pTreeNode->m_pLeft && !pTreeNode->m_pRight); 21 if(currentSum == expectedSum && isLeaf) 22 { 23 std::vector<int>::iterator iter = path.begin(); 24 for(; iter != path.end(); ++ iter) 25 std::cout << *iter << '\t'; 26 std::cout << std::endl; 27 } 28 29 // if the node is not a leaf, goto its children 30 if(pTreeNode->m_pLeft) 31 FindPath(pTreeNode->m_pLeft, expectedSum, path, currentSum); 32 if(pTreeNode->m_pRight) 33 FindPath(pTreeNode->m_pRight, expectedSum, path, currentSum); 34 35 // when we finish visiting a node and return to its parent node, 36 // we should delete this node from the path and 37 // minus the node's value from the current sum 38 currentSum -= pTreeNode->m_nValue; 39 path.pop_back(); 40 }
注:参数传引用是多余的。如果把传引用参数修改为传值参数,那么倒数第二行的currentSum -= pTreeNode->m_nValue;也需要去掉。这样的代码在效果上是一致的。
以上转自何海涛博客