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

Leetcode 113. Path Sum II

时间:2019-03-10 09:31:08      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:col   ==   etc   问题   tco   efi   none   app   return   

递归时要记录路径.

过程中遇到一个小问题,这两句:

a.insert(0,root.val)
ans.append(a)

一开始写成了这样:

ans.append(a.insert(0,root.val))

得到WrongAnswer.

因为a.insert()这个函数完成后不会返回任何值(None).

# 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):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: List[List[int]]
        """
        if not root:
            return []
        
        ans=[]
        if (not root.left)&(not root.right):
            if root.val==sum:
                ans.append([root.val])
            return ans
        elif not root.right:
            ans_left=self.pathSum(root.left,sum-root.val)
            if ans_left:
                for a in ans_left:
                    if a:
                        a.insert(0,root.val)
                        ans.append(a)
            return ans
        elif not root.left:
            ans_right=self.pathSum(root.right,sum-root.val)
            if ans_right:
                for a in ans_right:
                    if a:
                        a.insert(0,root.val)
                        ans.append(a)
            return ans
        else:
            ans_left=self.pathSum(root.left,sum-root.val)
            if ans_left:
                for a in ans_left:
                    if a:
                        a.insert(0,root.val)
                        ans.append(a)
            ans_right=self.pathSum(root.right,sum-root.val)
            if ans_right:
                for a in ans_right:
                    if a:
                        a.insert(0,root.val)
                        ans.append(a)
            return ans

 

Leetcode 113. Path Sum II

标签:col   ==   etc   问题   tco   efi   none   app   return   

原文地址:https://www.cnblogs.com/zywscq/p/10504071.html

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