二叉树中和为某一值的路径

来源:互联网 发布:青岛广电网络客服电话 编辑:程序博客网 时间:2024/06/05 11:20

题目描述:
  输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点 所经过的结点形成一条路径。

思路:
  当用前序遍历的方式访问到某一节点时,我们把这个节点添加到路径上,并累加该节点的值,如果该节点为叶子节点并且路径中节点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前节点不是叶节点,则继续访问它的子节点。当前节点访问结束后,递归函数将自动回到它的父节点。因此我们在函数退出之前要在路径上删除当前节点,并减去当前节点的值,以确保返回父节点时路径刚好是从根节点到父节点的路径。不难看出保存路径的数据结构实际上是一个栈,因为路径要与递归调用状态一致,而递归调用的本质就是一个压栈和出栈的过程。

程序:

//递归版本/*struct TreeNode {    int val;    struct TreeNode *left;    struct TreeNode *right;    TreeNode(int x) :            val(x), left(NULL), right(NULL) {    }};*/class Solution {public:    vector<vector<int> > listRes;    vector<int> listTemp;    void IsFind(TreeNode* node,int num){        listTemp.push_back(node->val);        if(0 == num - node->val && !node->left && !node->right){//判断在叶子节点处和是否相等            listRes.push_back(listTemp);        }else{            if(node->left) IsFind(node->left,num-node->val);            if(node->right) IsFind(node->right,num-node->val);        }        listTemp.pop_back(); //深度遍历完一条路径回退    }    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {        if(root) IsFind(root,expectNumber);        return listRes;    }};
0 0
原创粉丝点击