码迷,mamicode.com
首页 > 其他好文 > 详细

113. 路径总和 II-二叉树的中序遍历

时间:2020-07-05 17:10:36      阅读:50      评论:0      收藏:0      [点我收藏+]

标签:backtrack   solution   style   遍历   for   tps   中序遍历   return   def   

问题描述

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树,以及目标和 sum = 22,

5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:

[
[5,4,11,2],
[5,8,4,5]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii

解答

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

二叉树的中序遍历。

‘‘‘
class Solution(object):
    def pathSum(self, root, sum):
        if root == None:
            return []
        result = []
        def backtrack(root, arr, count):
            if root.left == None and root.right == None:
                arr.append(root.val)
                if count + root.val == sum:
                    result.append([])
                    result[-1].extend(arr)
                arr.pop()
                return
            if root.left != None:
                arr.append(root.val)
                count += root.val
                backtrack(root.left, arr, count)
                count -= root.val
                arr.pop()
            if root.right != None:
                arr.append(root.val)
                count += root.val
                backtrack(root.right, arr, count)
                count -= root.val
                arr.pop()
        backtrack(root, [], 0)
        return result

 

113. 路径总和 II-二叉树的中序遍历

标签:backtrack   solution   style   遍历   for   tps   中序遍历   return   def   

原文地址:https://www.cnblogs.com/xxxxxiaochuan/p/13246541.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!