题目:
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。二叉树结点的定义如下:
struct BinaryTreeNode{ int m_nValue; BinaryTreeNode* m_pLeft; BinaryTreeNode* m_pRight; }
思路:
先说一个具体的例子,输入下图的二叉树和整数22,则打印出两条路径,第一条路径包含结点10、12,第二条路径包含结点10、5和7。
上面的图具体的分析了这个示例。当用前序遍历的方式访问到某一结点时,我们把该结点添加到路径上,并累加该结点的值。如果该结点为叶结点并且路径中结点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到它的父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是从根节点到父结点的路径。从分析中得知我们需要借助栈来保存路径的数据。
代码实现:
public ArrayList<ArrayList<Integer>> findPath(BinaryTreeNode pRoot, int expectedSum){ ArrayList<ArrayList<Integer>> pathList = new ArrayList<ArrayList<Integer>>(); if(pRoot == null){ //如果此树为空则直接返回 return null; } Stack<Integer> path = new Stack<Integer>();//定义栈来存储一条路径 findPath(pRoot, expectedSum, path, pathList); //调用查找方法 return pathList; }
private void findPath(BinaryTreeNode pRoot, int expectedSum, Stack<Integer> path, ArrayList<ArrayList<Integer>> pathList) { if(pRoot == null){ return; } //判断其是不是叶子结点,如果是看其是不是等于期望值,是的话直接添加到ArrayList集合当中 if(pRoot.leftNode == null && pRoot.rightNode == null){ if(pRoot.value == expectedSum){ //如果等于期望值,则添加到list集合中 ArrayList<Integer> list = new ArrayList<Integer>(); for (int value : path) { //将路径上的值添加集合中 list.add(value); } list.add(pRoot.value);//将根节点值添加入集合 pathList.add(list); //将这条路径添加到pathList集合中 } }else{ //不是叶子结点,前序遍历,将当前结点值放入path栈中 path.push(pRoot.value); findPath(pRoot.leftNode, expectedSum - pRoot.value, path, pathList); findPath(pRoot.rightNode, expectedSum - pRoot.value, path, pathList); //在返回到父结点之前,在路径上删除当前结点的值 path.pop(); } }
小结:
有时候思路想不到时,就多举举例子。有时候有了思路,代码实现不了时,多学学基础与敲敲代码。这个题目考查树的先序遍历与栈的进出栈结合的题目。