JZ82 二叉树中和为某一值的路径(一)
版本2-利用递归调用,递归会遇到一个问题,如果想要计算某一条的路径值不好写,因为二叉树有两个方向,直接计算只能返回一个值,这里很巧妙用了减法,结果是知道的,看看每一条的路径值是否满足,自己就错在这里,一直纠结如何计算总的值。
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(root == nullptr)
{
return false;
}
if(root->left == nullptr && root->right == nullptr && sum - root->val == 0)
{
return true;
}
return hasPathSum(root->left, sum - root->val)||hasPathSum(root->right,sum - root->val);
}
};
版本2- 利用辅助栈深度优先遍历
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(root == NULL)
return false;
stack<pair<TreeNode*, int> > s;
s.push({root, root->val});
while(!s.empty()){
auto temp = s.top();
s.pop();
if(temp.first->left == NULL && temp.first->right == NULL && temp.second == sum)
return true;
if(temp.first->left != NULL)
s.push({temp.first->left, temp.second + temp.first->left->val});
if(temp.first->right != NULL)
s.push({temp.first->right, temp.second + temp.first->right->val});
}
return false;
}
};