0
点赞
收藏
分享

微信扫一扫

算法-二叉树:路径总和(是否存在路径)

boomwu 2022-02-07 阅读 105

算法-二叉树:路径总和(是否存在路径)

判断是否存在一条从根节点到叶子节点的路径,使这个路径的节点总和等于目标值。

//方法一:递归
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;
}
举报

相关推荐

0 条评论