给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
叶子节点 是指没有子节点的节点。
代码可直接放IDE运行
#include <iostream>
using namespace std;
// Definition for a binary tree node.
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
bool hasPathSum(TreeNode* root, int targetSum) {
if (root == nullptr) {
return false;
}
// 如果当前节点是叶子节点且值等于目标和,则返回 true
if (root->left == nullptr && root->right == nullptr && root->val == targetSum) {
return true;
}
// 递归检查左子树和右子树是否存在路径满足目标和
return hasPathSum(root->left, targetSum - root->val) || hasPathSum(root->right, targetSum - root->val);
}
int main() {
// 构造一个示例二叉树
TreeNode* root = new TreeNode(5);
root->left = new TreeNode(4);
root->right = new TreeNode(8);
root->left->left = new TreeNode(11);
root->left->left->left = new TreeNode(7);
root->left->left->right = new TreeNode(2);
root->right->left = new TreeNode(13);
root->right->right = new TreeNode(4);
root->right->right->right = new TreeNode(1);
int targetSum = 22;
if (hasPathSum(root, targetSum)) {
cout << "true" << endl;
} else {
cout << "false" << endl;
}
return 0;
}
时间复杂度:O(n),其中 n 是二叉树中节点的数量。在最坏情况下,我们需要遍历整个二叉树的节点才能确定是否存在根节点到叶子节点的路径满足目标和。
空间复杂度:O(h),其中 h 是二叉树的高度。在递归过程中,系统需要维护递归调用的栈空间。最坏情况下,二叉树完全不平衡,高度为 n,此时空间复杂度为 O(n)。最好情况下,二叉树完全平衡,高度为 log(n),此时空间复杂度为 O(log(n))。