简易Parser入门【一】：从树的递归开始

Parser就是将字符串转化成AST（Abstract Syntax Tree）的程序

首先咱们从树类的简单解析开始，假设有一个树节点（它是一个二叉树节点）：

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

咱们用这个树节点构造一棵树：

t_root = Tr(5, Tr(3, Tr(4), Tr(6)),
               Tr(1, Tr(2), Tr(7)))

它的结构如图所示，咱们想打印从根节点到每一个叶子节点的路径怎么办呢？遍历树通常都能想到用递归，可是递归到底递归什么东西呢？递归函数的参数如何选择？返回什么？

思考递归的时候，咱们能够从问题点出发，先寻找中止递归的条件。咱们须要打印从根节点到每一个叶子节点的路径，因此中止条件就是到达叶子节点了。能够先用代码写出来：

def print_leaf(tr_in):
    # type: (Tr) -> None
    if tr_in.left is None and tr_in.right is None:
        # todo: stop here and do something

咱们达到中止条件须要作什么呢？须要打印。打印什么？打印从根节点到叶子节点的路径。好了，咱们如今达到了叶子节点，函数拥有该节点tr_in的val，可是以前路径的值咱们尚未。咱们须要有，那么先假设咱们有一个，这个怎么来的呢，根据思考，以前路径的值固然是沿途遍历节点传进来的，那么天然须要当成函数输入传进来：

def print_leaf(tr_in, str_in):
    # type: (Tr, str) -> None
    if tr_in.left is None and tr_in.right is None:
        print(str_in + str(tr_in.val))

接着咱们思考，咱们没有到达叶子节点的时候，须要作什么。到达叶子节点是中止的标志，那么没有到达叶子节点就是递归的标志。咱们构造左节点或右节点还有叶子的状况：

def print_leaf(tr_in, str_in=""):
    # type: (Tr, str) -> None
    if tr_in.left is not None:
        print_leaf(tr_in.left, str_in + str(tr_in.val))

    if tr_in.right is not None:
        print_leaf(tr_in.right, str_in + str(tr_in.val))

    if tr_in.left is None and tr_in.right is None:
        print(str_in + str(tr_in.val))

这样整个代码就写好了，咱们实现了一个递归函数，它遇到有子节点的状况就递归调用本身，并一路把沿途的值存到str_in里边，到达最深层叶子的时候就中止，并把路径打印出来，完整代码以下：

class Tr:

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


def print_leaf(tr_in, str_in=""):
    # type: (Tr, str) -> None
    if tr_in.left is not None:
        print_leaf(tr_in.left, str_in + str(tr_in.val))

    if tr_in.right is not None:
        print_leaf(tr_in.right, str_in + str(tr_in.val))

    if tr_in.left is None and tr_in.right is None:
        print(str_in + str(tr_in.val))


if __name__ == "__main__":
    t_root = Tr(5, Tr(3, Tr(4), Tr(6)),
                   Tr(1, Tr(2), Tr(7)))
    print_leaf(t_root)

打印效果以下：

534
536
512
517