【程序员面试金典】面试题 04.12. 求和路径

题目描述

描述：给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

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

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

3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

节点总数 <= 10000

解题思路

思路1：最直观的想法是，回溯啊回溯！设置一个变量pathNum用于统计路径数目。首先由于路径不一定非得从二叉树的根节点开始，故需要遍历二叉树，求在以每一个节点为根节点的二叉树中满足从该二叉树根节点出发且目标和等于sum的路径数，此处可以使用前序遍历，边界条件是如果当前根节点为空则返回0。即PathSum是求以root为根节点的目标和为sum的路径数，其首先是判断从root出发的路径且目标和为sum的路径数，然后分别求以root的左(右)子树为根节点的目标和为sum的路径数。其中以root出发的路径且目标和为sum的路径数的做法，如果root为空则返回，反之将sum减去root值，然后分别在root的左子树和右子树中寻找目标和sum，如果root的左右子树均遍历完了，则再把root值加回sum。

int pathNum=0;
// 以某一节点为根节点的满足要求的路径和
void dfs(TreeNode* root,int sum)
{
   //到达叶子节点则返回
   if(root==NULL) return;
   //当前节点等于目标值则将路径树加一 注意此处不能返回因为可能存在负值
   if(root->val==sum) pathNum++;
   //减去当前值
   sum-=root->val;
   dfs(root->left,sum);
   dfs(root->right,sum);
   //左右都遍历完了则回溯
   sum+=root->val;
}
int pathSum(TreeNode* root, int sum) 
{
    if(root==NULL) return 0;
    dfs(root,sum);
    pathSum(root->left,sum);
    pathSum(root->right,sum);
    return pathNum;
}

总结：注意，以root为根节点的目标和为sum和以root出发的路径且目标和为sum这两者的概念不一样喔，以root为根节点的目标和为sum其不一定路径从root出发。该题和检查子树中的t2是t1的子树以及t2与以t1为根节点的二叉树相等这两个概念类似，如果是第一次写可能会有点绕，注意区分！