剑指刷题二叉树中和为某个整数的所有路径

最新推荐文章于 2021-12-23 13:06:47 发布

云治

最新推荐文章于 2021-12-23 13:06:47 发布

阅读量130

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/m0_37896011/article/details/99590828

数据结构专栏收录该内容

64 篇文章

订阅专栏

本文介绍了一种算法，用于查找二叉树中所有路径，这些路径上的节点值之和等于给定的整数。通过回溯法遍历每一条路径，记录路径及其总和，若到达叶子节点且路径总和等于目标值，则保存该路径。

题目描述

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

解题思路：按照回溯法的思路，遍历每一条路径，并记录路径与路径之和，如果遍历到叶子结点，路径之和不等于期望的数字，那么放弃这条路径，回到上一根节点，选择另一条路径。

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/

bool comp(const vector<int> & a, const vector<int> & b)
{
    return a.size() > b.size();
}// 在返回值的list中，数组长度大的数组靠前


#include <algorithm>
class Solution {
    private:
    vector<vector<int>> res;
    int my_expectNumber;
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        my_expectNumber=expectNumber;
        vector<int> arr;
        int num=0;
        is_path(root,arr,num);
        sort(res.begin(),res.end(),comp);// 在返回值的list中，数组长度大的数组靠前
        return res;
    }
    
    void is_path(TreeNode *root ,vector<int> arr,int num)//root记录当前结点，arr记录路径的值，num记录 路径之和
    {
        if(root==NULL)
            return ;
        arr.push_back(root->val);
        num+=root->val;
        if(root->left==NULL&&root->right==NULL)//如果是叶子结点
        {
            if(num==my_expectNumber)//并且 num是期望寻找的整数  将 这个数组压入返回值list中
            {
                res.push_back(arr);
                
            }
            return;
        }
        is_path(root->left,arr,num);//左子树递归
        is_path(root->right,arr,num);//右子树递归
        return;
    }
};