LeetCode | Path Sum

最新推荐文章于 2020-05-18 23:56:38 发布

原创最新推荐文章于 2020-05-18 23:56:38 发布 · 473 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #递归 #遍历

LeetCode 专栏收录该内容

16 篇文章

订阅专栏

本文探讨了如何确定一棵二叉树中是否存在从根到叶子节点的路径，使得这条路径上所有节点值之和等于给定的数值。通过递归算法实现路径求和的检查。

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 and

sum
 = 22

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

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

思路：本题意思是在二叉树中是否存在一条路径，该路径上的元素值总和为已知给出的sum。

因此，实际上可以将每层的结点值做一个转化，例如上述例子中第二层的结点值为4,8，那么到达第二层时，实际路径上的元素值和是5+4,5+8。即9,13.

所以，可以采用这种方法，逐层地下去，直到叶子节点。判断叶子结点值是否与sum相等即可。在程序中可以用递归的思路实现。

class Solution {
public:
    bool hasPathSum(TreeNode *root, int sum) 
	{
		if(!root)
			return false;
		else
			return PathSum(root,0,sum);
    }
	bool PathSum(TreeNode* p, int total, int sum)
	{
		if(!p->left && !p->right)
			return sum == total + p->val;
		else if(p->left && !p->right)
			return PathSum(p->left, total+p->val, sum);
		else if(p->right && !p->left)
			return PathSum(p->right, total+p->val, sum);
		else
			return (PathSum(p->left, total+p->val, sum) || PathSum(p->right, total+p->val, sum));
	}
};