剑指Offer算法题-二叉树中和为某一值的路径

发布于 / 刷题 / Comments Off on 剑指Offer算法题-二叉树中和为某一值的路径

题目描述

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

题解

dfs算法。注意路径要求是叶子节点,必须在dfs内判断是否为叶子节点,不能写成:

dfs(TreeNode* root){
if(root == NULL) 判断并return
dfs(root->left)
dfs(root->right)
}

这样会导致重复判断(一个叶子节点由于左右子树全为空,会两次进入dfs(NULL),导致两次进入root==NULL的逻辑内,导致重复判断)

除此之外,还要注意直接给一棵空树,root=NULL的情况。

/*
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> > result;

    void dfs(TreeNode* root, vector<int> & path, int expect, int sum = 0){
        sum += root->val;
        path.push_back(root->val);
        if(root->left)
            dfs(root->left, path, expect, sum);
        if(root->right)
            dfs(root->right, path, expect, sum);
        if(!root->left && !root->right)
            if(expect == sum) result.push_back(path);
        sum -= root->val;
        path.pop_back();
    }

    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<int> path;
        if(root)
            dfs(root, path, expectNumber);
        return result;
    }
};

转载原创文章请注明,转载自: 斐斐のBlog » 剑指Offer算法题-二叉树中和为某一值的路径
评论已关闭