剑指offer - 找路径

题目描述

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

思路

这个和之前做的leetcode中一个找所有可能的数字的组合很像。递归，不过记得加一下判断，是否走到了叶子节点

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def FindPath(self, root, target):
        anslist = []
        if root:
            if root.val < target:
                left = self.FindPath(root.left,target-root.val)
                right = self.FindPath(root.right,target-root.val)
                if left:
                    anslist += [[root.val]+ans for ans in left]
                if right>0:
                    anslist += [[root.val]+ans for ans in right]
                return anslist
            elif root.val == target:
                if root.left or root.right:
                    return []
                else:
                    return [[root.val]]
        else:
            return []

 

另一种思路

找到所有从根到叶子的路径，然后看路径加和是否为target。这里深度遍历和之前leetcode那道题很像。用三个变量

第一个变量 root

第二个变量 存放所有的paths，每次传递该数组的地址，直接修改数组内容

第三个变量 存放当前path， 一旦到达叶节点，把path存放到paths去。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def FindPath(self, root, target):
        if not root:
            return []
        paths = self.DFS(root,[],[root.val])
        ans = []
        for path in paths:
            if sum(path) == target:
                ans.append(path)
        return ans
    
    # 这里注意一下，之前写的不顺
    def DFS(self,root,result,path):
        if root.left == None and root.right==None:
            result.append(path)
        if root.left:
            self.DFS(root.left,result,path+[root.left.val])
        if root.right:
            self.DFS(root.right,result,path+[root.right.val])
        return result