面试题34:二叉树中和为某一值的路径(C++)

Posted wzw0625

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了面试题34:二叉树中和为某一值的路径(C++)相关的知识,希望对你有一定的参考价值。

题目地址:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof/

题目描述

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

题目示例

示例:
给定如下二叉树,以及目标和 sum = 22,

      5
      /     
    4        8
   /         /
  11      13  4
  /            /
 7   2        5  1
返回:

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

解题思路

思路1:首先分析题目,发现可以利用先序遍历实现,即“根元素->左子树->右子树”,在先序遍历实现时,我们使用path记录根节点到当前节点的路径,如果当前节点的路径中的各元素值恰好等于目标值sum时,则将此路径加入到结果集res中,递归左右子树,向上回溯。需要注意的是,如果要将此路径加入到结果集中的条件,即当前节点为叶子节点且路径各元素值恰好等于sum值。

思路2:利用后序遍历以及栈的方法解决,思路先序遍历差不多,需要注意的是后序遍历顺序是“左子树 ->根节点 ->右子树”。

程序源码

先序遍历+回溯(递归)

/**
 * 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 {
    vector<vector<int>> res; //存储决策路径
    vector<int> path; //存储决策元素值
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(root == nullptr) return {};
        prior_sort(root, sum);
        return res;
    }
    void prior_sort(TreeNode* root, int sum)
    {
        if(root == nullptr) return;
        path.push_back(root->val); //1.加入决策路径path
        sum -= root->val; //更新sum值
        if(sum == 0 && root->left == nullptr && root->right == nullptr) //2.找到可行解,加入结果集中
        {
            res.push_back(path);
        }
        //3.进入下一次决策
        prior_sort(root->left, sum);
        prior_sort(root->right, sum);
        path.pop_back(); //4.移出决策路径,即向上回溯,将当前节点从路径path中移出
    }
};

后序遍历+非递归

/**
 * 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) {
        if(root == nullptr) return {};
        stack<TreeNode*> sk;
        TreeNode* node = root, *prev_node = nullptr; //prev_node表示上一已被访问节点
        vector<vector<int>> res; //存储决策路径
        vector<int> path; //存储决策元素值
        int tmp_sum = 0;
        while(!sk.empty() || node != nullptr)
        {
            if(node != nullptr)
            {
                tmp_sum += node->val;
                path.push_back(node->val);
                sk.push(node);
                node = node->left; //一直往左遍历左子树
            }
            else
            {
                node = sk.top(); //不能往左了,取得根节点
                if(node->right == nullptr || node->right == prev_node)
                {
                    if(node->left == nullptr && node->right == nullptr && tmp_sum == sum)
                    {
                        res.push_back(path);
                    }
                    path.pop_back();
                    tmp_sum -= node->val;
                    sk.pop();
                    prev_node = node; //标记已使用
                    node = nullptr; //回溯
                }
                else
                {
                    node = node->right; //往右遍历右子树
                }
            }
        }
    return res;
    }
};

以上是关于面试题34:二叉树中和为某一值的路径(C++)的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode | 面试题34. 二叉树中和为某一值的路径剑指OfferPython

面试题34. 二叉树中和为某一值的路径(dfs)

剑指offer(C++)-JZ34:二叉树中和为某一值的路径(数据结构-树)

剑指offer(C++)-JZ34:二叉树中和为某一值的路径(数据结构-树)

剑指Offer对答如流系列 - 二叉树中和为某一值的路径

刷题记录-剑指offer34:二叉树中和为某一值的路径