【LeetCode】(112)Path Sum(Easy)

最新推荐文章于 2023-08-04 15:19:27 发布

原创最新推荐文章于 2023-08-04 15:19:27 发布 · 326 阅读

0 ·

CC 4.0 BY-SA版权

Algorithm 专栏收录该内容

81 篇文章

订阅专栏

本文探讨了在二叉树中查找从根节点到叶子节点的路径，使得路径上节点值之和等于指定目标值的问题。通过分析边界条件，简化了代码逻辑，实现了更高效的路径求和算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

Path Sum

Total Accepted: 65733 Total Submissions: 221541 My Submissions

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 - val

我的代码如下

/**
 * 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:

	 bool hasPathSumSub(TreeNode* root, int sum)
	 {
		 if (!root && 0 == sum)
		 {
			 return true;
		 }
		 else if (!root && sum != 0)
		 {
			 return false;
		 }


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

毫无疑问，搞的太繁琐了，其实边界条件分析不清楚是导致代码混乱的最主要原因，看了一眼大神的，自己修改后如下

/**
 * 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:

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

其实边界条件很简单，如果左右都没有节点，那就是叶子节点了，叶子节点的值看看是不是sum就ok了。