Path Sum(路径和)

版权声明:本文为博主原创文章,转载请注明出处。个人博客地址:https://yangyuanlin.club/ 欢迎来踩~~~~ https://blog.csdn.net/w7239/article/details/85634498

题目描述

Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.
For example:
Given the below binary tree andsum = 22,

return true, as there exist a root-to-leaf path5->4->11->2which sum is 22.

题目大意

给定一个二叉树和一个sum值,看看是否有一条从根到叶子结点的路径,使得这条路径上各个结点的值相加刚好等于sum。

思路

递归思路求解,递归每条路径,看是否有一条路径之和刚好等于sum。

递归出口是:
(1)返回为true:当前结点刚好是叶子结点&&sum-当前结点的值为0。
(2)返回为false:root == NULL。

避坑:sum和各个结点值有可能是负数;

代码

#include<iostream>
using namespace std;

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

typedef TreeNode* tree;

bool hasPathSum(TreeNode *root, int sum)
{
    if(root == NULL)return false;

    // 保证该店是叶子结点且加到这儿的和为sum
    if(root->val==sum && root->left==NULL && root->right==NULL)
        return true;

    return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
}

int main()
{
    tree t = new TreeNode(-2);
    t->left = new TreeNode(-3);
    cout<<hasPathSum(t, -5)<<endl;
    //cout<<-5-(-2)<<endl;
    return 0;
}

以上。

猜你喜欢

转载自blog.csdn.net/w7239/article/details/85634498