剑指offer 25.二叉树中和为某一值的路径

最新推荐文章于 2022-05-20 08:00:00 发布

快乐划水程序猿

最新推荐文章于 2022-05-20 08:00:00 发布

阅读量90

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/Alexia23/article/details/103843129

版权

刷题专栏收录该内容

174 篇文章 0 订阅

订阅专栏

题目：输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)
思路：
1. DFS遍历，到叶子结点之后检查路径和是否为sum，如果是sum加入结果集
启发或者坑：
1. 递归变量可以使用引用，极大节省空间

代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
 
 
class Solution {
public:
    int eN;
    vector<vector<int>> res;
    vector<vector<int>> FindPath(TreeNode* root, int expectNumber) {
        if (!root)
            return res;
        eN = expectNumber;
        vector<int> temp;
        int target = 0;
        FindAllPath(root, temp, target);
        return res;
    }
 
    void FindAllPath(TreeNode* root, vector<int>& path, int& sum) {
        //这个路径指的是从根节点到叶子几点的路径，遍历所有叶子节点的路径，如果等于eN，输出其路径
         
        path.push_back(root->val);
        sum += root->val;
        cout <<sum << endl;
        if (!root->left && !root->right) {
            if (sum == eN)
                res.push_back(path);     
            return;
        }
        if (root->left) {
            FindAllPath(root->left, path, sum);
            //cout << "start new leaf " <<endl;
            //cout << "before:" << sum << endl;
            sum -= path[path.size()-1];
            path.pop_back();
            //cout << "AFTER:" << sum << endl;
        }
        if (root->right) {
            FindAllPath(root->right, path, sum);
            sum -= path[path.size()-1];
            path.pop_back();
        }   
    }
};