使用python编写递归获取树形结构数据 & python寻找树指定节点位置路径

使用python编写递归获取树形结构数据

#  使用递归做一个常用的数据库的目录树结构递归,递归的数据如果太大容易将内存给吃光掉
import json

list_data = [
    {'id': 1, 'name': '体育0', 'pid': 0},  # pid为0表示顶级
    {'id': 2, 'name': '体育1', 'pid': 1},  # pid的值等于id，则表示是那个元素的子级
    {'id': 3, 'name': '体育2', 'pid': 1},
    {'id': 4, 'name': '体育3', 'pid': 2},
    {'id': 5, 'name': '体育4', 'pid': 2},
    {'id': 6, 'name': '体育5', 'pid': 5},
    {'id': 7, 'name': '体育6', 'pid': 5},
    {'id': 8, 'name': '体育7', 'pid': 5},
    {'id': 8, 'name': '体育7', 'pid': 11},
]


def get_list(pid):
    data = []

    for x in list_data:
        if x['pid'] == pid:
            next_pid = x['id']
            x['sons'] = get_list(next_pid)
            data.append(x)

    return data


if __name__ == '__main__':
    print(json.dumps(get_list(0), ensure_ascii=False))

python寻找树指定节点位置路径

import json

list_data = [
    {'id': 1, 'name': '体育0', 'pid': 0},  # pid为0表示顶级
    {'id': 2, 'name': '体育1', 'pid': 1},  # pid的值等于id，则表示是那个元素的子级
    {'id': 3, 'name': '体育2', 'pid': 1},
    {'id': 4, 'name': '体育3', 'pid': 2},
    {'id': 5, 'name': '体育4', 'pid': 2},
    {'id': 6, 'name': '体育5', 'pid': 5},
    {'id': 7, 'name': '体育6', 'pid': 5},
    {'id': 8, 'name': '体育7', 'pid': 5},
    {'id': 8, 'name': '体育7', 'pid': 11},
]


tree_dict = {}
for i in list_data:
    tree_dict[i['id']] = i['pid']
# 此时的tree_dict = {'2': '1', '3': '1', '4': '1', '5': '2', '6': '2', '7': '3'.......}

target = 2
res = []
while tree_dict.get(target):
    res.append(target)
    target = tree_dict.get(target)
res.append(target)
res.reverse()
print(res)