数据结构——树

2021-02-13 20:58:22 阅读：209 来源： 互联网

标签：branches tree branch return 数据结构节点 def

def  tree(root_label, branches = []):
    """定义一棵树，树根节点值，树的分支
    root_label: 根节点值
    branches: 分支，初始为空，只有根节点
    """
    for branch in branches:
#           遍历子树，每个子树都要是树
        assert is_tree(branch), 'branches must be trees'
#     返回根节点+子树列表
    return [root_label] + list(branches)


def label(tree):
    # 索引为0的为根节点
    return tree[0]

def branches(tree):
    #  索引1后为子树
    return tree[1:]

def is_tree(tree):
#     输入一颗树，判断是不是树, 即树需要是列表，并且长度大于1
    if type(tree) != list or len(tree) < 1:
        return False
    for branch in branches(tree):
#         检查子树是不是树
        if not is_tree(branch):
            return False
    return True

def is_leaf(tree):
#     判断一棵树是不是叶子节点， 如果不算子树，则是叶子节点
    return not branches(tree)
        
def fib_tree(n):
#     n：根节点值
    if n == 0 or n == 1:
        return tree(n)
    else:
        left, right = fib_tree(n - 2), fib_tree(n - 1)
#         n号节点值为n-1号和n-2号之和
        fib_n = label(left) + label(right)
#         tree(父节点值，子树)
        return tree(fib_n, [left, right])

def count_leaves(tree):
#     是叶子节点就返回1
    if is_leaf(tree):
        return 1
    else:
#     在每个子树里寻找叶子节点
        branch_counts = [count_leaves(branch) for branch in branches(tree)]
#     把子树里的叶子节点加起来
        return sum(branch_counts)

def partition_tree(n, m):
#     父节点值为0，树只有这一个节点，返回True
    if n == 0:
        return tree(True)
    elif n < 0 or m == 0:
        return tree(False)
    else:
#         左节点值为n - m
        left = partition_tree(n - m, m)
#         右节点值为m - 1 分割的n
        right = partition_tree(n, m - 1)
        return tree(m, [left, right])

def print_parts(tree, partition = []):
#     如果是叶子节点
    if is_leaf(tree):
#         如果该节点值不为0
#        递归基本条件：在叶子节点前加 + 输出， 
        if label(tree):
            print（'+'.join(partition))
    else:
        left, right = branches(tree)
        m = str(label(tree))
        print_parts(left, partition + [m])
        print_parts(right, partition)

def left_binarize(tree):
    if is_leaf(tree):
        return tree
    if len(tree) > 2:
        tree = [tree[1:],tree[0]]
    return (left_binarize(branch) for branch in tree)

left_binarize([1,2,3,4,5,6,7])

标签：branches,tree,branch,return,数据结构,节点,def
来源： https://blog.csdn.net/Ennis_Tongji/article/details/113803319

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

数据结构——树