数据结构|树遍历|问题2

简介

在树的遍历中，层序遍历是一种广泛应用的算法。但是，如果要按层级顺序将一颗二叉树展开成一个数组，该怎么做呢？这个问题就是本文要解决的。

解决方法

我们可以通过BFS的思路，依次遍历每一层节点，并将节点值按顺序存入数组中。

以下是Python实现代码：

def flatten_tree(root):
    if root is None:
        return []
    queue = [root]
    result = []
    while queue:
        n = len(queue)
        for i in range(n):
            node = queue.pop(0)
            result.append(node.val)
            if node.left:
                queue.append(node.left)
            if node.right:
                queue.append(node.right)
    return result

使用示例

我们来看一下下面这个二叉树：

     1
   /   \
  2     3
 / \   / \
4   5 6   7

按层级顺序展开的结果应该是 [1, 2, 3, 4, 5, 6, 7]。

我们可以先构建这个二叉树，然后调用上述代码进行验证：

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

root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)

root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

root.right.left = TreeNode(6)
root.right.right = TreeNode(7)

print(flatten_tree(root))   # [1, 2, 3, 4, 5, 6, 7]

总结

本文介绍了如何将一颗二叉树按层级顺序展开成数组。通过广度优先搜索的方式依次遍历每一层节点，并将节点值插入到数组中，最终得到我们要的结果。值得注意的是，这种方法也适用于多叉树。