无限级分类树状结构的应用场景很多，例如后端研发需要把用户相关权限读取出来并生成树状结构，前端研发拿到权限树之后可以按照结构展示用户有权限访问的栏目；再例如网页上的栏目分级： 作者在初次接触树状结构生成需求的时候，也是挠头，后来找到了一个代码少且清晰易懂的生成算法：递归。
首先，确保数据库中存储的类别信息如下：
[
{"id": 1, "name": ‘电器‘, "parent": 0},
{"id": 2, "name": ‘水果‘, "parent": 0},
{"id": 3, "name": ‘家用电器‘, "parent": 1},
{"id": 4, "name": ‘电吹风‘, "parent": 3},
{"id": 5, "name": ‘电风扇‘, "parent": 3},
{"id": 6, "name": ‘台灯‘, "parent": 3},
{"id": 7, "name": ‘商用电器‘, "parent": 1},
{"id": 8, "name": ‘大型电热锅‘, "parent": 7},
]
def generate_tree(source, parent):
tree = []
for item in source:
if item["parent"] == parent:
item["child"] = generate_tree(source, item["id"])
tree.append(item)
return tree
只需要将数据库中存储的信息传递给 generate_tree 函数即可。这段递归代码在往复循环的过程中通过 parent 来寻找子结点，找到子结点后将其添加到树中。完整代码如下：
import json
def generate_tree(source, parent):
tree = []
for item in source:
if item["parent"] == parent:
item["child"] = generate_tree(source, item["id"])
tree.append(item)
return tree
if __name__ == ‘__main__‘:
permission_source = [
{"id": 1, "name": ‘电器‘, "parent": 0},
{"id": 2, "name": ‘水果‘, "parent": 0},
{"id": 3, "name": ‘家用电器‘, "parent": 1},
{"id": 4, "name": ‘电吹风‘, "parent": 2},
{"id": 5, "name": ‘电风扇‘, "parent": 3},
{"id": 6, "name": ‘台灯‘, "parent": 3},
{"id": 7, "name": ‘商用电器‘, "parent": 1},
{"id": 8, "name": ‘大型电热锅‘, "parent": 7},
]
permission_tree = generate_tree(permission_source, 0)
print(json.dumps(permission_tree, ensure_ascii=False))
终端输出结果如下图所示：
递归算法中有很多重复的计算，这些计算不仅占用额外资源，还会降低函数执行效率，因此需要对递归进行优化。这里选用缓存优化法提升函数执行效率。
基本思路是每次找到结点关系后将此条目的编号添加到一个列表中缓存起来，代表此条目已找到结点关系。当往复循环执行函数时再次遇到此条目可以跳过。代码改动很简单，增加一个缓存列表和控制流语句即可：
def generate_tree(source, parent, cache=[]):
tree = []
for item in source:
if item["id"] in cache:
continue
if item["parent"] == parent:
cache.append(item["id"])
item["child"] = generate_tree(source, item["id"], cache)
tree.append(item)
return tree
至此，无限级分类树状结构生成算法完成。你学会了吗？
原文：https://www.cnblogs.com/laoyuming/p/14308448.html