构造无限级树的框架套路,附上python/golang/php/js实现

Posted 无涯

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了构造无限级树的框架套路,附上python/golang/php/js实现相关的知识,希望对你有一定的参考价值。

前言

框架思维非常重要,和语言无关,这是一种非常重要的抽象能力,吹得厉害一点,就是要有高屋建瓴,统筹全局的能力。
无限级树型结构的创建也是有套路的,下面由伟大的诗人chenqionghe给出套路框架,再分别套上相应的编程语言实现。
最终达到看到框架就能实现对应编程语言的代码,或者也可以直接copy实现拿去用~

迭代实现参考了这篇文章的思路列表list转树形结构:(python/golang/js/php),童鞋们可以看看这篇文章

需求

给定一级列表,构造出指定父ID的无限级树型结果

数据

[
  {\'id\': 1, \'parent_id\': 0, \'name\': "A"},
  {\'id\': 2, \'parent_id\': 1, \'name\': "AA"},
  {\'id\': 3, \'parent_id\': 1, \'name\': "AB"},
  {\'id\': 4, \'parent_id\': 3, \'name\': "ABA"},
  {\'id\': 5, \'parent_id\': 3, \'name\': "ABB"},
  {\'id\': 6, \'parent_id\': 3, \'name\': "ABC"},
  {\'id\': 7, \'parent_id\': 1, \'name\': "AC"},
  {\'id\': 8, \'parent_id\': 7, \'name\': "ACA"},
  {\'id\': 9, \'parent_id\': 8, \'name\': "ACAA"},
  {\'id\': 10, \'parent_id\': 8, \'name\': "ACAB"},
]

结果

最后构造成一棵根据父ID连接起来的树结构

{
    "id": 1,
    "parent_id": 0,
    "name": "A",
    "children": [
        {
            "id": 2,
            "parent_id": 1,
            "name": "AA",
            "children": []
        },
        {
            "id": 3,
            "parent_id": 1,
            "name": "AB",
            "children": [
                {
                    "id": 4,
                    "parent_id": 3,
                    "name": "ABA",
                    "children": []
                },
                {
                    "id": 5,
                    "parent_id": 3,
                    "name": "ABB",
                    "children": []
                },
                {
                    "id": 6,
                    "parent_id": 3,
                    "name": "ABC",
                    "children": []
                }
            ]
        },
        {
            "id": 7,
            "parent_id": 1,
            "name": "AC",
            "children": [
                {
                    "id": 8,
                    "parent_id": 7,
                    "name": "ACA",
                    "children": [
                        {
                            "id": 9,
                            "parent_id": 8,
                            "name": "ACAA",
                            "children": []
                        },
                        {
                            "id": 10,
                            "parent_id": 8,
                            "name": "ACAB",
                            "children": []
                        }
                    ]
                }
            ]
        }
    ]
}

框架

递归框架

递归是一门艺术,用接近人类的语言来表达了程序,优点是代码较少,缺点是性能较差。

获取树(列表,父ID)
   res = []
   for 节点 in 列表:
        if 节点的parent_id 等于 父ID
            节点.children = 获取树(列表, 节点ID)
            res.add(节点)
   return res

迭代框架

迭代本身是创建一条引用链,将所有的节点串起来

获取树(列表,父ID)
   memo = {}
   for 节点 in 列表:
        //构造memo给节点的父ID查找追加节点用
        if 节点ID in memo:
            节点.children = memo[节点ID].children //之前构造的children数组覆盖当前节点的children
            memo[节点ID] = 节点
        else
            节点.children = []
            memo[节点ID] = 节点
        
        //给像父对象的children追加
        if 节点父ID in memo:
            memo[节点父ID].children.add(memo[节点ID]) //追加当前构造的ID节点
        else:
            memo[节点父ID] = {\'children\':[memo[节点ID]]} //初始化父对象再追加
            
   return memo[父ID].children

递归框架实现

python

def get_tree_iterative(list_data, parent_id=0):
    memo = {}
    for v in list_data:
        item_id = v[\'id\']
        item_paren_id = v[\'parent_id\']
        if item_id in memo:
            v[\'children\'] = memo[item_id][\'children\']
            memo[item_id] = v
        else:
            v[\'children\'] = []
            memo[item_id] = v

        if item_paren_id in memo:
            memo[item_paren_id][\'children\'].append(memo[item_id])
        else:
            memo[item_paren_id] = {\'children\': memo[item_id]}
    return memo[parent_id][\'children\']


list_data = [
    {\'id\': 1, \'parent_id\': 0, \'name\': "A"},
    {\'id\': 2, \'parent_id\': 1, \'name\': "AA"},
    {\'id\': 3, \'parent_id\': 1, \'name\': "AB"},
    {\'id\': 4, \'parent_id\': 3, \'name\': "ABA"},
    {\'id\': 5, \'parent_id\': 3, \'name\': "ABB"},
    {\'id\': 6, \'parent_id\': 3, \'name\': "ABC"},
    {\'id\': 7, \'parent_id\': 1, \'name\': "AC"},
    {\'id\': 8, \'parent_id\': 7, \'name\': "ACA"},
    {\'id\': 9, \'parent_id\': 8, \'name\': "ACAA"},
    {\'id\': 10, \'parent_id\': 8, \'name\': "ACAB"},
]

res = get_tree_iterative(list_data)

import json

print(json.dumps(res, indent=4))

运行如下

golang

package main

import (
    "encoding/json"
    "fmt"
)

type Node struct {
    Id       int     `json:"id"`
    ParentId int     `json:"parent_id"`
    Name     string  `json:"name"`
    Children []*Node `json:"children"`
}

func getTreeRecursive(list []*Node, parentId int) []*Node {
    res := make([]*Node, 0)
    for _, v := range list {
        if v.ParentId == parentId {
            v.Children = getTreeRecursive(list, v.Id)
            res = append(res, v)
        }
    }
    return res
}

func main() {
    list := []*Node{
        {4, 3, "ABA", nil},
        {3, 1, "AB", nil},
        {1, 0, "A", nil},
        {2, 1, "AA", nil},
        {5, 3, "ABB", nil},
        {6, 3, "ABC", nil},
        {7, 1, "AC", nil},
        {8, 7, "ACA", nil},
        {9, 8, "ACAA", nil},
        {10, 8, "ACAB", nil},
    }
    res := getTreeRecursive(list, 0)
    bytes, _ := json.MarshalIndent(res, "", "    ")
    fmt.Printf("%s\\n", bytes)
}

运行如下

php

function getTreeRecursive(&$list, $parentId = 0)
{
    $res = [];
    foreach ($list as $k => $v) {
        if ($v[\'parent_id\'] == $parentId) {
            $v[\'children\'] = getTreeRecursive($list, $v[\'id\']);
            $res[] = $v;
        }
    }
    return $res;
}
$list = [
    [\'id\' => 4, \'parent_id\' => 3, \'name\' => "ABA"],
    [\'id\' => 3, \'parent_id\' => 1, \'name\' => "AB"],
    [\'id\' => 1, \'parent_id\' => 0, \'name\' => "A"],
    [\'id\' => 2, \'parent_id\' => 1, \'name\' => "AA"],
    [\'id\' => 5, \'parent_id\' => 3, \'name\' => "ABB"],
    [\'id\' => 6, \'parent_id\' => 3, \'name\' => "ABC"],
    [\'id\' => 7, \'parent_id\' => 1, \'name\' => "AC"],
    [\'id\' => 8, \'parent_id\' => 7, \'name\' => "ACA"],
    [\'id\' => 9, \'parent_id\' => 8, \'name\' => "ACAA"],
    [\'id\' => 10, \'parent_id\' => 8, \'name\' => "ACAB"],
];
$res = getTreeRecursive($list);
echo json_encode($res, JSON_PRETTY_PRINT);

运行结果如下

js

function getTreeRecursive(listData, parentId = 0) {
  let res = []
  listData.forEach((v, k) => {
    if (v.parent_id == parentId) {
      v.children = getTreeRecursive(listData, v.id)
      res.push(v)
    }
  })
  return res
}


dataList = [
  {\'id\': 1, \'parent_id\': 0, \'name\': "A"},
  {\'id\': 2, \'parent_id\': 1, \'name\': "AA"},
  {\'id\': 3, \'parent_id\': 1, \'name\': "AB"},
  {\'id\': 4, \'parent_id\': 3, \'name\': "ABA"},
  {\'id\': 5, \'parent_id\': 3, \'name\': "ABB"},
  {\'id\': 6, \'parent_id\': 3, \'name\': "ABC"},
  {\'id\': 7, \'parent_id\': 1, \'name\': "AC"},
  {\'id\': 8, \'parent_id\': 7, \'name\': "ACA"},
  {\'id\': 9, \'parent_id\': 8, \'name\': "ACAA"},
  {\'id\': 10, \'parent_id\': 8, \'name\': "ACAB"},
]

let res = getTreeRecursive(dataList);
console.log((JSON.stringify(res, null, 4)))

运行如下

迭代框架实现

python

def get_tree_iterative(list_data, parent_id=0):
    memo = {}
    for v in list_data:
        id_ = v[\'id\']
        item_paren_id = v[\'parent_id\']
        if id_ in memo:
            v[\'children\'] = memo[id_][\'children\']
            memo[id_] = v
        else:
            v[\'children\'] = []
            memo[id_] = v

        if item_paren_id in memo:
            memo[item_paren_id][\'children\'].append(memo[id_])
        else:
            memo[item_paren_id] = {\'children\': memo[id_]}
    return memo[parent_id][\'children\']


list_data = [
    {\'id\': 1, \'parent_id\': 0, \'name\': "A"},
    {\'id\': 2, \'parent_id\': 1, \'name\': "AA"},
    {\'id\': 3, \'parent_id\': 1, \'name\': "AB"},
    {\'id\': 4, \'parent_id\': 3, \'name\': "ABA"},
    {\'id\': 5, \'parent_id\': 3, \'name\': "ABB"},
    {\'id\': 6, \'parent_id\': 3, \'name\': "ABC"},
    {\'id\': 7, \'parent_id\': 1, \'name\': "AC"},
    {\'id\': 8, \'parent_id\': 7, \'name\': "ACA"},
    {\'id\': 9, \'parent_id\': 8, \'name\': "ACAA"},
    {\'id\': 10, \'parent_id\': 8, \'name\': "ACAB"},
]

res = get_tree_iterative(list_data)

import json

print(json.dumps(res, indent=4))

运行如下

golang

package main

import (
    "encoding/json"
    "fmt"
)

type Node struct {
    Id       int     `json:"id"`
    ParentId int     `json:"parent_id"`
    Name     string  `json:"name"`
    Children []*Node `json:"children"`
}

func getTreeIterative(list []*Node, parentId int) []*Node {
    memo := make(map[int]*Node)
    for _, v := range list {
        if _, ok := memo[v.Id]; ok {
            v.Children = memo[v.Id].Children
            memo[v.Id] = v
        } else {
            v.Children = make([]*Node, 0)
            memo[v.Id] = v
        }
        if _, ok := memo[v.ParentId]; ok {
            memo[v.ParentId].Children = append(memo[v.ParentId].Children, memo[v.Id])
        } else {
            memo[v.ParentId] = &Node{Children: []*Node{memo[v.Id]}}
        }
    }
    return memo[parentId].Children

}

func main() {
    list := []*Node{
        {4, 3, "ABA", nil},
        {3, 1, "AB", nil},
        {1, 0, "A", nil},
        {2, 1, "AA", nil},
        {5, 3, "ABB", nil},
        {6, 3, "ABC", nil},
        {7, 1, "AC", nil},
        {8, 7, "ACA", nil},
        {9, 8, "ACAA", nil},
        {10, 8, "ACAB", nil},
    }
    res := getTreeIterative(list, 0)
    bytes, _ := json.MarshalIndent(res, "", "    ")
    fmt.Printf("%s\\n", bytes)
}

运行如下

php

function getTreeIterative($list, $parentId = 0)
{
    $memo = [];
    foreach ($list as &$v) {
        $id = $v[\'id\'];
        $itemParentId = $v[\'parent_id\'];
        if (isset($memo[$id])) {
            $v[\'children\'] = &$memo[$id][\'children\'];
            $memo[$id] = $v;
        } else {
            $v[\'children\'] = [];
            $memo[$id] = $v;
        }
        if (isset($memo[$itemParentId])) {
            $memo[$itemParentId][\'children\'][] = &$memo[$id];
        } else {
            $memo[$itemParentId] = [\'children\' => [&$memo[$id]]];
        }
    }
    return $memo[$parentId][\'children\'];
}


$list = [
    [\'id\' => 4, \'parent_id\' => 3, \'name\' => "ABA"],
    [\'id\' => 3, \'parent_id\' => 1, \'name\' => "AB"],
    [\'id\' => 1, \'parent_id\' => 0, \'name\' => "A"],
    [\'id\' => 2, \'parent_id\' => 1, \'name\' => "AA"],
    [\'id\' => 5, \'parent_id\' => 3, \'name\' => "ABB"],
    [\'id\' => 6, \'parent_id\' => 3, \'name\' => "ABC"],
    [\'id\' => 7, \'parent_id\' => 1, \'name\' => "AC"],
    [\'id\' => 8, \'parent_id\' => 7, \'name\' => "ACA"],
    [\'id\' => 9, \'parent_id\' => 8, \'name\' => "ACAA"],
    [\'id\' => 10, \'parent_id\' => 8, \'name\' => "ACAB"],
];
$res = getTreeIterative($list);
echo json_encode($res, JSON_PRETTY_PRINT);

运行结果如下

js

function getTreeIterative(listData, parentId = 0) {
  let memo = {};
  listData.forEach((v, k) => {
    let id = v.id
    let itemParentId = v.parent_id

    if (memo[id]) {
      v.children = memo[id].children
      memo[id] = v
    } else {
      v.children = []
      memo[id] = v;
    }

    if (memo[itemParentId]) {
      memo[itemParentId].children.push(memo[id]);
    } else {
      memo[itemParentId] = {children: [memo[id]]};
    }
  })

  return memo[parentId].children
}


dataList = [
  {\'id\': 1, \'parent_id\': 0, \'name\': "A"},
  {\'id\': 2, \'parent_id\': 1, \'name\': "AA"},
  {\'id\': 3, \'parent_id\': 1, \'name\': "AB"},
  {\'id\': 4, \'parent_id\': 3, \'name\': "ABA"},
  {\'id\': 5, \'parent_id\': 3, \'name\': "ABB"},
  {\'id\': 6, \'parent_id\': 3, \'name\': "ABC"},
  {\'id\': 7, \'parent_id\': 1, \'name\': "AC"},
  {\'id\': 8, \'parent_id\': 7, \'name\': "ACA"},
  {\'id\': 9, \'parent_id\': 8, \'name\': "ACAA"},
  {\'id\': 10, \'parent_id\': 8, \'name\': "ACAB"},
]

let res = getTreeIterative(dataList);
console.log((JSON.stringify(res, null, 4)))

运行如下

以后如果想用直接来拷贝代码就行,就是这么简单~

以上是关于构造无限级树的框架套路,附上python/golang/php/js实现的主要内容,如果未能解决你的问题,请参考以下文章

JSON数据格式生成无限级树结构

php通用的树型类创建无限级树型菜单

144. 二叉树的前序遍历

算法框架套路找出给定数据中所有的树

二叉树的递归套路

jstl处理递归问题+jquery,tree无限级树形菜单+简单实用案例