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 的路径有:
- 5 -> 3
- 5 -> 2 -> 1
- -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++用户)