LeetCode 刷题【112. 路径总和】

题目描述

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

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

示例:

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

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

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

题目来源：力扣（LeetCode）
题目链接：https://leetcode-cn.com/problems/path-sum

解题思路

用递归遍历树中每一条从根节点到叶子节点的路径：参数为（root->child, sum-root->val）
1. 当节点为非叶子节点时：递归遍历孩子节点；
2. 当节点为叶子节点时：判断（sum-当前节点的值val）是否为0：
- 为0说明该树中存在根节点到该叶子节点的路径，这条路径上所有节点值相加等于目标和sum，程序返true；
- 不为0说明该路径不符合要求；继续递归遍历下一条路径；
- 如果遍历完所有路径都没有一条路径符合，则程序返回false。

关键代码

/**
 * 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 ret = false;
    bool hasPathSum(TreeNode* root, int sum) {
        //递归遍历头结点到尾结点；每遍历一个节点，用sum减去当前节点值，
        //并用该差作为新的sum递归遍历该节点的子节点；
        if(root == NULL) return ret;//当根节点为空时
        if(root->right == NULL && root->left == NULL && (sum-root->val)==0){
            ret = true;
            return ret;//找到一条符合题意的路径，程序返回
        }
        if(root->left != NULL)
            hasPathSum(root->left, sum-root->val);//递归遍历左孩子
        if(root->right != NULL)
            hasPathSum(root->right, sum-root->val);//递归遍历右孩子
        return ret;
    }
};

具体代码见题解链接