[每日一题]103:二叉树中和为某一值的路径

124 阅读1分钟

文章目录


题目描述

给定一个二叉树和一个值 sum,判断 是否有从根节点到叶子节点的节点值之和等于 sum 的路径,

例如:

给出如下的二叉树,sum=22,

在这里插入图片描述

返回true,因为存在一条路径 5→4→11→2的节点值之和为 22

示例1

输入
{1,2},0

输出
false

示例2

输入
{1,2},3

输出
true

题解思路

递归

实现代码:

bool hasPathSum(TreeNode* root, int sum) {
    if (nullptr == root) return false;
    else if (root->val == sum && root->left == nullptr && root->right == nullptr) {
        return true;
    }
    return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
}

题目描述

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

示例:

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

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

返回:

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

题解思路

/**
 * 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>> res;
        vector<int> cur;
        dfs(root, sum, res, cur);
        return res;
    }

    void dfs(TreeNode* root, int sum, vector<vector<int>>& res, vector<int>& cur){
        if(!root) return;
        cur.push_back(root->val);
        sum -= root->val;
        if(sum == 0 && !root->left && !root->right) res.push_back(cur); // 满足路径条件

        dfs(root->left,  sum, res, cur);
        dfs(root->right, sum, res, cur);
        cur.pop_back(); // 关键点:回溯
    }
};

如有不同见解,欢迎留言讨论~~~