[python]层次遍历+中序遍历重建二叉树

Description：
给定二叉树的层次及中序遍历，重建二叉树，输出该树的前序遍历及叶节点。
输入两行，分别为层次遍历及中序遍历。
输出两行，分为为叶节点及前序遍历。
Example:
Input:
3 5 4 2 6 7 1
2 5 3 6 4 7 1
Output:
2 6 1
3 5 2 4 6 7 1

---

思路： 对于子树，其在层次遍历最前面的点即为根节点，故重建过程包括以下两步：
1.利用层次遍历确定子树的根节点；
2.根据根节点在中序中的位置划定左右子树，递归重建。

代码：

class TreeNode:  # 定义二叉树节点类
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

def restruct(level, mid):
    if not level or not mid:
        return None
    lst = []
    for num in mid:
        lst.append(level.index(num))
    idx = lst.index(min(lst))
    root = TreeNode(mid[idx])
    left = mid[0:idx]
    right = mid[idx + 1:]
    root.left = restruct(level, left)
    root.right = restruct(level, right)
    return root

lst1 = []
lst2 = []

def pre_traverse(root):
    if not root:
        return None
    lst1.append(root.val)
    pre_traverse(root.left)
    pre_traverse(root.right)
    return lst1

def leaf(root):
    if not root:
        return None
    if not root.left and not root.right:
        lst2.append(root.val)
    leaf(root.left)
    leaf(root.right)
    return lst2

b = restruct([3, 5, 4, 2, 6, 7, 1], [2, 5, 3, 6, 4, 7, 1])
print(leaf(b))
print(pre_traverse(b))

输出：

---

类似：前序遍历+中序遍历重建二叉树：牛客OJ

思路： 重建过程包括以下递归的以下两步：
1.前序遍历的第一个节点为子树的根节点；
2.根据根节点在中序中的位置划定左右子树，递归重建。

代码：

class TreeNode:  # 定义二叉树节点类
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
        
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        if not pre or not tin:
            return None
        root=TreeNode(pre[0])
        idx=tin.index(pre[0])
        left=tin[0:idx]
        right=tin[idx+1:]
        root.left=self.reConstructBinaryTree(pre[1:len(left)+1],left)
        root.right=self.reConstructBinaryTree(pre[-len(right):],right)
        return root