剑指 34 二叉树中和为某一值的路径【回溯】【递归】

剑指 34 二叉树中和为某一值的路径【回溯】【递归】

全部刷题与学习记录

【C++刷题学习笔记目录】

【C++百万并发网络通信-笔记目录】

原题目

题目地址：https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof/

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

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

---

考查知识点

回溯、递归

---

自己的第一遍解法

设计到二叉树的题目十有八九离不开递归，题目中设计到求和，那么就需要一个累加器count，返回的是一个二维数组finalRes，二维数组中的每个一维数组onceVec代表一条路径。力扣上对于路径没有详细解释，可能是认为大家都懂。

从树的根节点开始一直往下到叶节点所经过的节点形成一条路径

我写了一个递归函数能够找到一个满足条件的一维数组，但是怎么继续查找多条路径就没了思路。

class Solution_1 {//不完善，只能打印出一个结果，并且没有考虑路径的结束
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if (root == nullptr)    return vector<vector<int>> {};

        int count = 0;
        vector<int> onceVector = {};
        vector<vector<int>> finalVector;
        bool flag = recursion(root, count, sum, onceVector);
        if (flag)
        {
            finalVector.push_back(onceVector);
        }

        return finalVector;
    }
    bool recursion(TreeNode* root, int count, int val, vector<int>& onceRes)
    {
        count += root->val;
        if (count < val)
        {
            onceRes.push_back(root->val);
            if ((root->left && recursion(root->left, count, val, onceRes)) ||
                    root->right && (recursion(root->right, count, val, onceRes)))
                return true;
        }
        if (count > val)    return false;

        onceRes.push_back(root->val);
        return true;//count == val
    }
};

这里的不完善主要就是：1、没有考虑路径的完整性，如果是路径还没有结束，当前路径中的节点数值已经满足条件，这段代码也能通过。2、只能找到一条（不完善）路径，不具备多条路径的查找能力

---

好的解法

在力扣题解https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof/solution/zi-jie-ti-ku-jian-34-zhong-deng-er-cha-shu-zhong-h/，我找到了这个思路的升级版。

class Solution {
private:
    vector<vector<int> > finalRes;
    vector<int> onceVec;//一条路径
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(!root) return {};
        recursion(root, sum);
        return finalRes;
    }
    void recursion(TreeNode *root, int sum)
    {
        if(!root) return;
        onceVec.push_back(root -> val);//进入当前节点就将节点加入
        sum -= root -> val;
        if(sum == 0 && !root -> left && !root -> right)
            finalRes.push_back(onceVec);//满足条件时，只是将当前路径添加到最终结果中
            //查找出一条路径后不能退出，要让递归函数自己退出，这样才会遍历全部节点
        recursion(root -> left, sum); // 左
        recursion(root -> right, sum); // 右
        onceVec.pop_back(); // 函数退出之前先弹出当前节点
    }
};

1、通过使用类成员，避免了我一开始向递归函数频繁传入数组的麻烦

2、在递归函数进入一个节点时，就将该节点加入当前路径中，若在当前节点已经满足要求 and 当前节点是叶节点（这就满足了一条路径的完整要求）的情况下，将当前路径添加到结果中，注意，这里仅仅是添加，并没有直接返回该路径。我在一开始的思路中，正是受到了“满足条件就要返回”的错误影响，导致只能查找一条路径。

最后是使递归函数进入两个叶节点，这也是保障所有节点都能被访问的操作。

在递归函数退出当前节点时，意味：已经完成了当前节点的所有操作，需要返回父节点的位置，这时一定要将当前节点从当前路径中删除