leetcode 112. 路径总和(DFS)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/kirito0104/article/details/82078129

给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例: 
给定如下二叉树，以及目标和 sum = 22，

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

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

思路

1、一开始想在hasPathSum中得到各子树的和，发现不行，原因是函数只能返回一个值，那是返回左子树的和还是有字数的和呢？

2、看了别人的code，发现非常巧妙，他是在getSum中判断的，然后返回判断结果。此外还用一个形参来保存目前的和

3、简单总结：垂直计算（路径和）在子程序中判断，横向比对（最大深度）可在主程序中判断

code

/**
 * 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 0;
        return getSum(root,0,sum);
    }
    bool getSum(TreeNode* root,int sum,int a){
        if(!root && sum == a)
            return 1;
        if(!root && sum != a)
            return 0;
        if(root->left && !root->right) return getSum(root->left,sum+root->val,a);
        if(!root->left && root->right) return getSum(root->right,sum+root->val,a);
        return getSum(root->left,sum+root->val,a)||getSum(root->right,sum+root->val,a);
    }
};