LeetCode构建树的测试用例

最近几天有些佛了

由于一直赶导师分配的项目

所以LeetCode没怎么刷了

开始接触到树的题目了

一直卡在了如何构建一颗树的测试用例

所以想写篇文章专门分享一下

对了，分享一波最近做的书签

---

首先分享如何构建一颗二叉树的代码吧，因为这种情况比较特殊，所以更好写。

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


# 创建一二叉树
def creatBTree(data, index):
    pNode = None
    if index < len(data):
        if data[index] == None:
            return
        pNode = TreeNode(data[index])
        pNode.left = creatBTree(data, 2 * index + 1)
        pNode.right = creatBTree(data, 2 * index + 2)
    return pNode

if __name__ == "__main__":
    data = [1, None, 2, 3]
    pNode = creatBTree(data, 0)

再给出如何构建一普通树的代码，至于是怎么构建大家自行理解吧，不难。

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

# 创建树
def build(data):
    if len(data) == 0:
        return TreeNode(0)
    nodeQueue = []
    # 创建一根节点，并将根节点进栈
    root = TreeNode(data[0])
    nodeQueue.append(root)
    # 记录当前行节点的数量
    lineNum = 2
    # 记录当前行中数字在数组中的位置
    startIndex = 1
    # 记录数组中剩余元素的数量
    restLength = len(data) - 1
    while restLength > 0:
        for index in range(startIndex, startIndex + lineNum, 2):
            if index == len(data):
                return root
            cur_node = nodeQueue.pop()
            if data[index] is not None:
                cur_node.left = TreeNode(data[index])
                nodeQueue.append(cur_node.left)
            if index + 1 == len(data):
                return root
            if data[index + 1] is not None:
                cur_node.right = TreeNode(data[index + 1])
                nodeQueue.append(cur_node.right)
        startIndex += lineNum
        restLength -= lineNum
        # 此处用来更新下一层树对应节点的最大值
        lineNum = len(nodeQueue) * 2
    return root

if __name__ == "__main__":
    data = [1, None, 2, 3]
    pNode = build(data)

给出建立树的代码了，再顺带也把遍历树的三种方法给写出来。这三种方法无非也就是先序遍历，中序遍历以及后序遍历，其具体概念这儿就不讲了，百度一下很快就知道了。我直接给代码。

# 先序遍历二叉树
def preTraverse(root):
    if root is None:
        return
    print(root.val)
    preTraverse(root.left)
    preTraverse(root.right)


# 中序遍历二叉树
def midTraverse(root):
    if root is None:
        return
    midTraverse(root.left)
    print(root.val)
    midTraverse(root.right)


# 后序遍历二叉树
def afterTraverse(root):
    if root is None:
        return
    afterTraverse(root.left)
    afterTraverse(root.right)
    print(root.val)