数据结构学习 jz34 二叉树中和为某一值的路径

关键词：回溯 二叉树 前序遍历 路径记录

因为我没有仔细接触过二叉树的遍历过程，所以我是懵懵懂懂按照dfs的方法写的。没想到写对了，看了解答发现这叫做二叉树的前序遍历。用时29min。

这让我明白了前序遍历和dfs原来是有相同之处的。（我甚至想按照习惯给它剪枝，后来发现不太行，每条路都必须走一遍才行，我似乎又懂得了许多呢！）

题目：

思路：

注意这里是每个节点都要访问到的，不能剪枝！！！

容器：

得弄两个vector。

std::vector<std::vector<int>> res：记录最后要返回的结果。

std::vector<int> temp：记录中途的路径，如果有符合的路径，就把这个vector push到res里面。

sum记录中途的结果。

注意：函数形参里的std::vector<int>& temp不要漏引用，不然内存会多用很多！

中止条件：

到达叶子节点就停： （别剪枝）

 if (root->left == nullptr && root->right == nullptr)

逐个查询：

这里有两个要查询的东西，左节点和右节点。

if (root->left != nullptr)
    dfs(root->left, target, temp, sum + root->val);
if (root->right != nullptr)
    dfs(root->right, target, temp, sum + root->val);

复杂度计算：

时间复杂度O(n)

空间复杂度O(n) 主要是统计temp用掉的额外空间，如果数退化成链表，就需要n

代码：

#include <vector>
struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode() : val(0), left(nullptr), right(nullptr) {}
     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 };

class Solution {
public:
    std::vector<std::vector<int>> res;
    std::vector<std::vector<int>> pathTarget(TreeNode* root, int target) {
        if (root == nullptr) return res;
        std::vector<int> temp;
        dfs(root, target, temp, 0);
        return res;
    }
    void dfs(TreeNode* root, int target, std::vector<int>& temp, int sum)
    {
        temp.push_back(root->val);
        if (root->left == nullptr && root->right == nullptr)//如果叶子节点
        {
            if (sum + root->val == target)//如果符合
            {
                res.push_back(temp);
            }
        }
        else
        {
            if (root->left != nullptr)
                dfs(root->left, target, temp, sum + root->val);
            if (root->right != nullptr)
                dfs(root->right, target, temp, sum + root->val);
        }
        temp.pop_back();//回溯
        return;
    }
    
};