n叉树结构的检索算法实现

浅浅的花香味﹌ 2022-12-02 13:34 82阅读 0赞

使用python实现n叉树检索算法。  
n叉树结构基于现实业务，基本数据结构如下：

[
      {
        
        "id": "",
        "name": "公司1",
        "children": [
          {
        
            "id": "",
            "name": "公司2",
            "children": []
          }
        ]
      }
    ]

实现代码如下，使用了两个函数进行递归调用

import json
    
    
    def find_text(obj, text):
        """
        检索该对象及子对象的名称中是否含有指定关键字
        父节点有关键字，返回True
        如没有，检索子节点，一旦发现匹配则返回True
        如果父子节点都检索不到，返回False
        :param obj: dict, 节点对象
        :param text: str, 关键字
        :return:boolean
        """
        if text in obj["name"]:
            return True
        else:
            for z_obj in obj["children"]:
                if find_text(z_obj, text):
                    return True
        return False
    
    
    def push_obj(obj_list, text):
        """
        将符合条件的节点对象存入新的列表中
        遍历所有对象，并判断对象及其子对象名称中是否匹配关键字
        如匹配，则存入新的列表，否则不做任何动作
        与此同时判断子节点是否匹配，如果匹配一并放入新列表
        :param obj_list: list, 节点对象列表
        :param text: str, 关键字
        :return: list, 节点对象列表
        """
        new_list = []
    
        for obj in obj_list:
            if find_text(obj, text):
                s_list = push_obj(obj["children"], text)
                obj["children"] = s_list
                new_list.append(obj)
    
        return new_list
    
    
    def main():
        # 如果需要文件可以发邮件联系
        with open("tree.json", "r", encoding="utf-8") as f:
            obj_list = json.load(f)
        text = "5"
        result = push_obj(obj_list, text)
        
        for r in result:
            print(r)
    
    
    if __name__ == '__main__':
        main()