`

程序员面试题精选(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();  
分享到:
评论

相关推荐

    程序员面试100题数据结构和算法

    4. **二元树路径和**:找出所有和为目标值的二元树路径。可以使用深度优先搜索(DFS)或广度优先搜索(BFS),同时跟踪路径上的节点值,当路径和等于目标值时记录路径。 5. **最小k个元素**:快速找到数组中的最小k...

    C/C++程序员面试指南.杨国祥(带详细书签).pdf

    面试题15:在二元树中找出和为某一值的所有路径 第11章 排序 11.1 插入排序 面试题1:编码实现直接插入排序 面试题2:编码实现希尔(Shell)排序 11.2 交换排序 面试题3:编码实现冒泡排序 面试题4:编码实现快速...

    刀疤鸭之数据结构面试题

    5. 二元树中找出和为某一值的所有路径: 这个问题可以通过深度优先搜索(DFS)来解决,需要递归地在树中寻找所有可能的路径,直到找到总和等于目标值的路径。 ***K算法: TopK算法涉及到在一组数中找出前k个最大...

    c和c++500强面试题

    5. 在二元树中找出和为某一值的所有路径:这是一个递归遍历二元树并记录路径和的问题。可以通过深度优先搜索(DFS)来实现,在递归过程中累加路径上的节点值,当达到叶子节点且路径和等于目标值时,记录该路径。 **...

    常见:算法面试题21

    【算法面试题21】是针对程序员面试时常见的算法题目集合,主要考察的是思维能力和算法应用。以下将详细解析这些题目: 1. **二元查找树转双向链表**: 这个问题要求将一棵二元查找树(BST)转换成一个有序的双向...

    微软编程面试100题

    【微软编程面试100题】是一系列针对程序员面试的挑战题目,涵盖了数据结构和算法等领域,旨在评估候选人的编程能力和逻辑思维。以下是其中几道题目的解析: 1. **二元查找树转双向链表**: 这道题要求将一个二元...

Global site tag (gtag.js) - Google Analytics