剑指offer-二叉树中和为某一值的路径C++实现

阿吉的CV之路

于 2021-12-12 13:08:14 发布

阅读量745

点赞数

分类专栏： leetcode刷题总结文章标签： c++ 深度优先 leetcode 剑指offer

本文链接：https://blog.csdn.net/qq_35140742/article/details/121885897

版权

leetcode刷题总结专栏收录该内容

27 篇文章 0 订阅

订阅专栏

剑指offer-二叉树中和为某一值的路径C++实现

原题链接

在这里插入图片描述

#include <iostream>
#include <vector>

using namespace std;

struct TreeNode {
    TreeNode *left;
    TreeNode *right;
    int val;

    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    vector<vector<int>> FindPath(TreeNode *root, int expectNumber) {
        if (!root)return {};
        // 先序遍历
        dfs(root, expectNumber);
        return ans;
    }

    // 一维数组用于存放每个目标路径节点
    vector<int> tmp;

    // 先序遍历搜索符合要求的节点
    void dfs(TreeNode *root, int x) {
        // 越过叶子节点时，回溯
        if (!root) return;
        // 储存每个遍历过的节点值
        tmp.emplace_back(root->val);
        // 寻找符合要求的路径
        x -= root->val;
        // 走到目标路径的叶节点时，将路径节点的一维数组存到结果数组中
        if (!root->left && !root->right && x == 0) {
            ans.emplace_back(tmp);
        }
        // 先左再右
        dfs(root->left, x);
        dfs(root->right, x);
        // 回溯前删除该节点，避免存储重复和错误的节点值
        tmp.pop_back();
    }

private:
    vector<vector<int>> ans;
};