【剑指OFFER】二叉树中和为某一值的路径

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

MakaloLi

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

阅读量83

点赞数

分类专栏：剑指

本文链接：https://blog.csdn.net/qq_33532867/article/details/87539089

版权

剑指专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该博客介绍了如何在二叉树中找到所有节点值之和等于给定整数的路径。通过深度优先搜索策略遍历二叉树，确保路径从根节点延伸至叶节点，并且只有在到达叶节点时才判断路径和是否等于目标值。

摘要由CSDN通过智能技术生成

题目描述

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

算法设计

用深度搜索，逐个遍历二叉树的结点。要注意的是题目中说，路径是从树的根结点开始往下一直到叶节点，因此如果还未到叶节点，值的和就与expectNumber相等的话是不能算做一个正确结果的。

C++代码

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    void func(TreeNode *root, int expectNumber, set<vector<int> > &res_set, vector<int> nums){
        vector<int> res = nums;
        if(root == nullptr){
            return;
        }
        int exp = expectNumber - root->val;
        res.push_back(root->val);
        if(exp < 0){
            return;
        }
        else if(exp == 0){
            if(root->left == nullptr && root->right == nullptr)
                res_set.insert(res);
            return;
        }
        else{
            func(root->left, exp, res_set, res);
            func(root->right, exp, res_set, res);
        }
    }
    
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        set<vector<int> > res_set;
        vector<vector<int> > res;
        vector<int> nums;
        func(root, expectNumber, res_set, nums);
        set<vector<int> >::iterator it = res_set.begin();
        for(it;it!=res_set.end();it++){
            res.push_back(*it);
        }
        return res;
    }
};