剑指 Offer 34. 二叉树中和为某一值的路径

最新推荐文章于 2022-04-07 13:16:23 发布

GridWorker

最新推荐文章于 2022-04-07 13:16:23 发布

阅读量114

点赞数

分类专栏：剑指Offer

本文链接：https://blog.csdn.net/lyric12138/article/details/107748518

版权

剑指Offer 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

该博客介绍了如何在给定的二叉树中找到所有节点值之和等于特定目标值的路径。提供了示例和相关链接，帮助理解问题并给出解决方案。

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

剑指 Offer 34. 二叉树中和为某一值的路径

题目描述
代码实现

题目描述

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例：
给定如下二叉树，以及目标和 sum = 22，

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

返回：

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

提示：

节点总数 <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof

代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<vector<int>> result;
        if(!root) return result;
        vector<vector<int>> candidate = collectPath(root);       
        for(auto cit = candidate.begin(); cit != candidate.end(); cit++){
            int sumtmp = 0;
            for(auto ccit = (*cit).begin(); ccit != (*cit).end(); ccit++){
                sumtmp += (*ccit);
            }
            if(sum == sumtmp)
                result.push_back((*cit));
        }
        return result;
    }
    vector<vector<int>> collectPath(TreeNode* root) {
        vector<vector<int>> result;
        if((!root->left) && (!root->right)){
            vector<int> nodevec;
            nodevec.push_back(root->val);
            result.push_back(nodevec);
        }
        if(root->left){
            vector<vector<int>> leftPathes = collectPath(root->left);
            for(auto pait = leftPathes.begin(); pait != leftPathes.end(); pait++){
                (*pait).insert((*pait).begin(), root->val);
            result.push_back((*pait));
            }
        }
        if(root->right){
            vector<vector<int>> rightPathes = collectPath(root->right);
            for(auto pait = rightPathes.begin(); pait != rightPathes.end(); pait++){
                (*pait).insert((*pait).begin(), root->val);
                result.push_back((*pait));
            }
        }
        return result;
    }
};