算法-二叉树:路径总和(是否存在路径)
判断是否存在一条从根节点到叶子节点的路径,使这个路径的节点总和等于目标值。
//方法一:递归
bool pathSum(TreeNode* root,int target){
if(root == NULL) return false;
if(!root->left && !root->right && target==root->val) { //遇到叶子节点
return true;
}
return pathSum(root->left, target - root->val) || pathSum(root->right, target - root->val);
}
//方法二:迭代
bool pathSum(TreeNode* root,int target){
if(root == NULL) return false;
stack<pair<TreeNode*, int>> st; //栈既要记录从头节点到叶子节点的指针,又要记录该路径的和
st.push(pair<TreeNode*,int>(root,root->val));
while (!st.empty()) {
pair<TreeNode*, int> node = st.top();
st.pop();
if(!node.first->left && !node.first->right && target==node.second) { //遇到叶子节点
return true;
}
if(node.first->right){//右
st.push(pair<TreeNode*, int>(node.first->right,node.second + node.first->right->val));
}
if(node.first->left){//左
st.push(pair<TreeNode*, int>(node.first->left,node.second + node.first->left->val));
}
}
return false;
}