给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2
。
这道题刚拿到没什么思路,想着似乎只能暴力求解。
在网上搜索了一下大神的算法,虽然只是简单的递归调用,可是隔着屏幕我都想为这个巧妙的递归鼓掌!
话不多说,贴代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
bool hasPathSum(struct TreeNode* root, int sum)
{
if(root==NULL)
return false;
int t=sum-root->val;
if(root->left==NULL&&root->right==NULL)
return t==0?true:false;
return hasPathSum(root->left,t)||hasPathSum(root->right,t);
}
每次调用都令t=sum-root->val,再判断是否为0。t=0则为true,反之为false。