给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
解题思路:看到这道题,首先觉得与二叉树的所有路径非常像,完全可以用二叉树的所有路径思路来解题。需要做修改的地方就是在获取路径的时候进行求和,最后所有路径和与目标和做比较。
代码:
class Solution { //112. 路径总和
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if (root == nullptr) return false;
vector<int> allPathSum;
_getPathSum(root, 0, allPathSum);
for (int i = 0; i < allPathSum.size(); ++i) {
if (allPathSum[i] == targetSum) {
return true;
}
}
return false;
}
private:
void _getPathSum(TreeNode* cur, int sum, vector<int>& allPathSum) {
sum += cur->val;
if (cur->left == nullptr && cur->right == nullptr) {
allPathSum.push_back(sum);
return;
}
if (cur->left) {
_getPathSum(cur->left, sum, allPathSum);
}
if (cur->right) {
_getPathSum(cur->right, sum, allPathSum);
}
}
};
进阶方法:当我看了代码随想录的解题方法,发现了我之前写的方法的问题。其实当遍历时,找到与目标和相同的路径值,就应该直接返回,我之前的方法是对所有路径都进行了遍历。
代码:
class Solution { //112. 路径总和 进阶
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if (root == nullptr) return false;
if (root->left == nullptr && root->right == nullptr && targetSum == root->val) return true;
return hasPathSum(root->left, targetSum - root->val) || hasPathSum(root->right, targetSum - root->val);
}
};