本文共 951 字,大约阅读时间需要 3 分钟。
Q:输入一颗二叉树和一个整数,打印出二叉树中结点和为输入整数的所有路径。从树的根结点开始到叶子节点所经过的结点形成一条路径。结点定义如下:
struct BinaryTreeNode{ int m_nValue; BinaryTreeNode *m_pLeft; BinaryTreeNode *m_pRight;}这个问题直白点就是二叉树的遍历,在遍历的过程中要随时记录一些数据,在数据满足一定条件时,做一些特定的动作。
我们很自然的想到二叉树的三种遍历,以先根(前序)遍历来说,其遍历的递归模板是:
void PreOrder(BinaryTreeNode *root){ if(!root) return; visit(root); PreOrder(root->m_pLeft); PreOrder(root->m_pRight);}真对不同的要求visit()可以做不同的处理,这里的要求是判断当前的结点和是否等于给定的值,同时形成了一条路径(为叶子结点),若是,则输出这条路径。
void FindPath(BinaryTreeNode *pRoot, int expecteSum, int curSum){ if(!pRoot) return; curSum += pRoot->m_nValue; path.push_back(pRoot); if(curSum == expecteSum && IsLeaf(pRoot)) { std::vector::iterator iter = path.begin(); while (iter != path.end()) { std::cout << (*iter)->m_nValue << " "; ++iter; } std::cout << std::endl; } if(pRoot->m_pLeft) FindPath(pRoot->m_pLeft,expecteSum,curSum); if(pRoot->m_pRight) FindPath(pRoot->m_pRight,expecteSum,curSum); path.pop_back();}
转载地址:http://grlji.baihongyu.com/