面试题25:二叉树中和为某一值的路径

来源:互联网 发布:淘宝客服成功案例 编辑:程序博客网 时间:2024/05/17 22:12

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


解:
带记忆的DFS


/*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> > FindPath(TreeNode* root,int expectNumber) {        vector<vector<int>> result;        if(!root)            return result;        vector<int> trace;        find(root, expectNumber, trace, result);            return result;    }    void find(TreeNode* root, int expectNumber, vector<int>& trace, vector<vector<int>>& result){        trace.push_back(root->val);        if(!root->left && !root->right){            if(root->val == expectNumber)                result.push_back(trace);        }        if(root->left){            find(root->left, expectNumber - root->val, trace, result);        }        if(root->right){            find(root->right, expectNumber - root->val, trace, result);        }        trace.pop_back();    }};
0 0