FBI树Python题解

题目描述

我们可以把由 “0” 和 “1” 组成的字符串分为三类：全 “0” 串称为 B 串，全 “1” 串称为 I 串，既含 “0” 又含 “1” 的串则称为 F 串。

FBI树是一种二叉树，它的结点类型也包括 F 结点，B 结点和 I 结点三种。由一个长度为 2^N2N 的 “01” 串 S 可以构造出一棵 FBI 树 T，递归的构造方法如下：

1. T 的根结点为 R，其类型与串 S 的类型相同；

2. 若串 S 的长度大于 1，将串 S 从中间分开，分为等长的左右子串 S1 和 S2 ；由左子串 S1 构造 R 的左子树 T1，由右子串 S2 构造 R 的右子树 T2。

现在给定一个长度为 2^N2N 的 “01” 串，请用上述构造方法构造出一棵FBI树，并输出它的后序遍历序列。

输入描述

第一行是一个整数 N（0 \leq N \leq 10）N（0≤N≤10）。

第二行是一个长度为 2^N2N 的 “01” 串。

输出描述

输出一个字符串，即 FBI 树的后序遍历序列。

输入输出样例

示例 1

输入

3
10001011

输出

IBFBBBFIBFIIIFF

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

解答

class Tree():
    def __init__(self, data=None, left=None, right=None):
        self.data = data
        self.left = left
        self.right = right

    def postorder(self):
        if self.left is not None:
            self.left.postorder()
        if self.right is not None:
            self.right.postorder()
        if self.data is not None:
            print(self.data, end='')

    def bfs(self):
        if self is None:
            return
        queue = [self]
        while queue:
            cur = queue.pop()  # 弹出元素
            print(cur.data,end=' ')  # 打印元素值
            if cur.left is not None:
                queue.append(cur.left)
            if cur.right is not None:
                queue.append(cur.right)


    def add_data(self, data):
        newnode = Tree(data)
        queue = [self]

        while queue:
            node = queue.pop(0)
            if node.left is None:
                node.left = newnode
                # print(node.data+'的左子树是'+newnode.data)
                return
            else:
                queue.append(node.left)
            if node.right is None:
                node.right = newnode
                # print(node.data+'的右子树是'+newnode.data)
                return
            else:
                queue.append(node.right)



def decideFBI(str):
    if '0' in str and '1' in str:
        return 'F'
    if '0' in str:
        return 'B'
    if '1' in str:
        return 'I'
    # return str


if __name__ == '__main__':
    N = int(input())
    str = input()

    tree = Tree(decideFBI(str))
    if len(str) > 1:
        queue = [str]
        while queue:
            i = queue.pop(0)
            if len(i) > 1:
                str1 = i[:int(len(i) / 2)]
                str2 = i[int(len(i) / 2):]
                tree.add_data(decideFBI(str1))
                # tree.bfs()
                # print('')
                tree.add_data(decideFBI(str2))
                # tree.bfs()
                # print('')
                queue.append(str1)
                queue.append(str2)

    tree.postorder()