C++——二叉树路径总和leetcode437

题目描述

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

思路

来源leetcode官方解答一，假设rootSum(p, val)是二叉树中以p为起点的路径和为val的路径的数目，则rootSum(p, val) = rootSum(p->left, val-root->val) + rootSum(p->right, val-root->val)，而路径总和即为以二叉树中所有节点为起始的rootSum的总和。因此在迭代外部还需要外层迭代。同时需要判断如果当前节点的值正好为所需要的，则返回+1。

代码

long pathSum(TreeNode* root, long target)
{
	if(!root) return 0;
	return rootSum(root, target)+pathSum(root->left, target)+pathSum(root->right, target);
}
long rootSum(TreeNode* root, long target)
{
	if(!root) return 0;
	if(root->val == target) return rootSum(root->left, 0) + rootSum(root->right, 0) + 1;
	return rootSum(root->left, target-root->val) + rootSum(root->right, target-root->val);
}