Leetcode 112：路径总和

题目描述

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

示例

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

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

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

代码实现

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

}

代码性能

执行用时：12ms（击败93.51%的C++用户）
内存消耗：19.8MB（击败50.80%的C++用户）

Leetcode 113：路径总和II

题目描述

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。说明: 叶子节点是指没有子节点的节点。

示例

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

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

返回:

[
[5,4,11,2],
[5,8,4,5]
]

代码实现

void hasPathSum(TreeNode* root, int sum, vector<vector<int>>& result, vector<int>& temp) {
	sum -= root->val;
	temp.push_back(root->val);
	if (sum == 0 && (root->left == nullptr) && (root->right == nullptr))
	{
		result.push_back(temp);
		return;
	}
	if (root->left)
	{
		hasPathSum(root->left, sum, result, temp);
		temp.pop_back();
	}
	if (root->right)
	{
		hasPathSum(root->right, sum, result, temp);
		temp.pop_back();
	}
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
	vector<vector<int>>result;
	if (root == nullptr)
		return result;
	vector<int>temp;
	hasPathSum(root, sum, result, temp);
	return result;
}

代码性能

执行用时：12ms（击败98.02%的C++用户）
内存消耗：19.7MB（击败88.70%的C++用户）

Leetcode 437：路径总和III

题目描述

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

  10
 /  \
5   -3

/ \
3 2 11
/ \
3 -2 1

返回 3。和等于 8 的路径有:

1. 5 -> 3
2. 5 -> 2 -> 1
3. -3 -> 11

代码实现

int psum(TreeNode* root, int sum)
{
	if (root == nullptr)
		return 0;
	int res = 0;
	if (root->val == sum)
		res += 1;
	res += psum(root->left, sum - root->val);
	res += psum(root->right, sum - root->val);
	return res;
}
int pathSum(TreeNode* root, int sum) {
	if (root == nullptr)
		return 0;
	return psum(root, sum) + pathSum(root->left, sum) + pathSum(root->right, sum);
}

代码性能

时间复杂度
O(n)，n为树的节点个数；
空间复杂度
O(h)，h为树的高度；

执行用时：40ms（击败56.55%的C++用户）
内存消耗：14.5MB（击败89.02%的C++用户）