剑指offer:二叉树中和为某一值的路径
admin
2023-07-08 05:02:42
0

题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    # 返回二维列表,内部每个列表表示找到的路径
    """
    根据题目定义,一个路径总是从根节点开始的,而搜索路径本质上又和树的遍历有关,那么我们可以思考常
    用的树的遍历中哪种遍历方法可能有效,在这里答案就是先序遍历,因为先序遍历也是从根节点开始的。
    由于题目要求搜索所有可能的路径,一个路径在一层里面只经过一个节点,那么就是深度优先搜索DFS。

    具体到路径的定义,需要从根节点到叶节点,因此每遍历到一个节点的时候,如果这个节点是叶节点,那么
    我们就应该判断当前的路径是否符合要求,而如果这个节点不是叶节点,那么就选择一个子节点继续遍历。
    当一个节点的所有子节点都已经访问完毕之后,由于递归的特性会自动返回到其父节点,因此我们需要手动
    从路径中删除这个节点
    """
    def FindPath(self, root, expectNumber):
        def helper(cur_root, path):
            # 遍历到一个节点的时候先将其加入路径中
            path.append(cur_root.val)
            # 如果这个节点是叶节点,那么判断当前路径是否符合要求
            if not cur_root.left and not cur_root.right:
                if sum(path) == expectNumber:
                    # 这里注意不能res.append(path),因为path只是一个id,绑定了内存中的
                    # 一个列表,而我们后面还会修改这个列表,因此需要将当前路径的一个拷贝加到
                    # 返回结果当中
                    res.append(path[:])

            # 如果当前节点不是叶节点,那么访问其子节点
            if cur_root.left:
                helper(cur_root.left, path)
            if cur_root.right:
                helper(cur_root.right, path)

            # 在访问完这个节点的所有子节点之后会返回到其父节点中,
            # 在返回之前应该在路径中删除这个节点
            path.pop(-1)

        if not root:
            return []

        res = []
        helper(root, [])
        return res

def main():
    root = TreeNode(10)
    root.left = TreeNode(5)
    root.right = TreeNode(12)
    root.left.left = TreeNode(4)
    root.left.right = TreeNode(7)

    solution = Solution()
    print(solution.FindPath(root, 22))

if __name__ == '__main__':
    main()

相关内容

热门资讯

我国科学家为细胞信号“导航”开... 新华社济南5月31日电(记者张力元)人体细胞犹如一座精密的通信城市,每天都有大量“指令”穿梭传递,调...
极端大风突袭哈尔滨!过山车停摆... 极目新闻记者 詹钘5月31日,受强对流天气影响,哈尔滨国际会展中心体育场相关设施受到损坏,原计划当晚...
三原电缆取得电缆接头连接用防护... 国家知识产权局信息显示,上海三原电缆附件有限公司取得一项名为“一种电缆接头连接用防护结构”的专利,授...
原创 识... 还是那句话,机圈苦大屏久已…… 虽然大屏有大屏的美,但是小屏也有小屏的俏。在大屏旗舰占据主流的手机市...
玄戒技术取得分频电路专利,实现... 国家知识产权局信息显示,北京玄戒技术有限公司取得一项名为“分频电路、分频器、射频芯片和电子设备”的专...
为什么今年香会基调明显变了 5月29日—31日在新加坡举行的第23届香格里拉对话会(简称“香会”),见证着元首引领下大国关系继续...
成本几毛钱、假驱蚊液香精兑水,... 入夏升温,蚊虫进入活跃期,驱蚊防护成为民生刚需,《财经调查》持续接到消费者投诉,他们买到的多款网红驱...
越来越多80后90后,正在丧失... 六一儿童节到来之际,朋友圈里开始出现一种熟悉的热闹。有人晒出零食礼包,有人半开玩笑地向伴侣讨礼物,还...
洋保电子取得用于低温环境的电气... 国家知识产权局信息显示,洋保电子(太仓)有限公司取得一项名为“一种用于低温环境的电气柜”的专利,授权...
中日韩飞手争霸宁波!2026无... 潮新闻客户端 记者 陈冲 通讯员 朱凝 5月31日,2026小遛·无人机竞速世界杯(中国·宁波鄞州站...