112. 路徑總和

112. 路徑總和

題目

使用廣度優(yōu)先搜索的方式,記錄從根節(jié)點(diǎn)到當(dāng)前節(jié)點(diǎn)的路徑和。
使用兩個(gè)隊(duì)列,分別存儲(chǔ)將要遍歷的節(jié)點(diǎn),以及根節(jié)點(diǎn)到這些節(jié)點(diǎn)的路徑和即可。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if root==None:
            return False
        nodestack = [root]
        sumstack = [root.val]
        while nodestack:
            p = nodestack.pop(0)
            cursum = sumstack.pop(0)
            
            if p.left:
                cur_left_sum = cursum + p.left.val
                nodestack.append(p.left)
                sumstack.append(cur_left_sum)
            if p.right:
                cur_right_sum = cursum + p.right.val
                nodestack.append(p.right)
                sumstack.append(cur_right_sum)

            if not p.left and not p.right:
                if cursum==sum:
                    return True
        return False
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容