数据结构 python：树简单操作

最新推荐文章于 2023-12-22 09:08:27 发布

伟菜

最新推荐文章于 2023-12-22 09:08:27 发布

阅读量421

点赞数

分类专栏： Python 文章标签：数据结构 python

本文链接：https://blog.csdn.net/wei_bo_cai/article/details/116025077

版权

本文介绍了如何使用Python实现树数据结构的基本操作，包括初始化树、层级遍历、前序遍历、后序遍历以及查找节点等。代码简洁易懂，适用于数据结构的学习和实践。

摘要由CSDN通过智能技术生成

数据结构：树的生成

- - 原始数据
  - 代码

笔者提供的代码，都是基于python的语言特性实现的，简单明了为上

原始数据

child: 子节点
parent：父节点
如果parent==child，则为根节点（这里是森林的结构）

child,parent
a,b
b,c
c,c

代码

初始化树
层级遍历
前序遍历、后序遍历
获取枝干，查找节点

import json

from copy import deepcopy
from collections import defaultdict, OrderedDict


class Tree:
    def __init__(self, par2chi: list, parent_key='parent', child_key='child'):
        """
        有父子结构组成的数据体系，结束标志：obj[self.parent_key] == obj[self.child_key]
        :param par2chi: [{self.parent_key: '', self.child_key: ''}]
        """
        self.child_key = child_key
        self.parent_key = parent_key
        self.tree = self.init_tree(par2chi)

    def __end_signal(self, parent, child):
        """
        结束信号，森林的根节点的标志
        """
        return parent == child

    def init_tree(self, par2chi):
        """
        初始化树结构，主要是利用 python 的语言特性实现的，表面上看，时间复杂度是n，空间复杂度n，n为par2chi的大小
        """
        original = defaultdict(dict)
        original_c = defaultdict(dict)
        for row in par2chi:
            if row[self.child_key] not in original_c:
                original_c[row[self.child_key]] = dict()
            if self.__end_signal(row[self.parent_key], row[self.child_key]):
                original[row[self.parent_key]] = original_c[row[self.child_key]]
            else:
                original_c[row[self.parent_key]][row[self.child_key]