给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/
4 8
/ /
11 13 4
/
7 2 1
返回 true
, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2
。
class Solution { bool res; public: bool hasPathSum(TreeNode* root, int sum) { if(!root) return false; pathSum(root, sum); return res; } void pathSum(TreeNode* root, int sum) {
//当前节点为叶节点且值等于sum时表示找到路径 if(!root->left && !root->right && root->val == sum) { res = true; return; }
//当前节点为叶节点但不等于sum时直接返回 if(!root->left && !root->right) return;
//当前为非叶节点时继续左右子树的递归 //if(root->val != sum) { //不能加上这句判断,因为可能从根节点到某非叶节点的路径和就已经等于sum了,而题目要求根节点到叶节点路径 if(root->left) pathSum(root->left, sum - root->val); if(root->right) pathSum(root->right, sum - root->val);
//} } };