`

程序员面试题精选(04)-在二元树中找出和为某一值的所有路径

阅读更多

题目:输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。
例如输入整数22和如下二元树
                                             10
                                            /    \
                                           5      12
                                         /    \  
                                      4      7 
则打印出两条路径:10, 12和10, 5, 7。
二元树结点的数据结构定义为:

C++代码 复制代码
  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. };  
struct BinaryTreeNode // a node in the binary tree
{
      int               m_nValue; // value of node
       BinaryTreeNode   *m_pLeft;  // left child of node
       BinaryTreeNode   *m_pRight; // right child of node
};

分析:这是百度的一道笔试题,考查对树这种基本数据结构以及递归函数的理解。
当访问到某一结点时,把该结点添加到路径上,并累加当前结点的值。如果当前结点为叶结点并且当前路径的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是根结点到父结点的路径。我们不难看出保存路径的数据结构实际上是一个栈结构,因为路径要与递归调用状态一致,而递归调用本质就是一个压栈和出栈的过程。
参考代码:

C++代码 复制代码
  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.        currentSum += pTreeNode->m_nValue;   
  15.        path.push_back(pTreeNode->m_nValue);   
  16.       // if the node is a leaf, and the sum is same as pre-defined,    
  17.       // the path is what we want. print the path   
  18.       bool isLeaf = (!pTreeNode->m_pLeft && !pTreeNode->m_pRight);   
  19.       if(currentSum == expectedSum && isLeaf)   
  20.        {       
  21.             std::vector<int>::iterator iter = path.begin();   
  22.            for(; iter != path.end(); ++ iter)   
  23.                  std::cout << *iter << '\t';   
  24.            std::cout << std::endl;   
  25.        }   
  26.   
  27.       // if the node is not a leaf, goto its children   
  28.       if(pTreeNode->m_pLeft)   
  29.              FindPath(pTreeNode->m_pLeft, expectedSum, path, currentSum);   
  30.       if(pTreeNode->m_pRight)   
  31.              FindPath(pTreeNode->m_pRight, expectedSum, path, currentSum);   
  32.   
  33.       // when we finish visiting a node and return to its parent node,   
  34.       // we should delete this node from the path and    
  35.       // minus the node's value from the current sum   
  36.        currentSum -= pTreeNode->m_nValue;   
  37.        path.pop_back();  
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics