美文网首页
一维数组转树,无限层级,代码已优化,可直接使用

一维数组转树,无限层级,代码已优化,可直接使用

作者: 剑指流云 | 来源:发表于2021-08-01 15:28 被阅读0次

经过几次优化,已经达到了我所能达到的最优解,特写出代码优化的过程,记录自己的成长,想复制代码的直接跳转文末,本方法在循环过程中添加了索引,不需要的可自行删除

  • 例如:
1.-----
  1-1.-----
  1-2.---
2.---
  2-1.---
    2-1-1---
    2-1-2---
    2-1-3---
  2-2.---
    2-2-1---

路漫漫其修远兮,吾将上下而求索。

树中循环是避免不了的,只能尽量减少少循环,代码仅供交流,并不是最优解

  • 方法是用vue实现的,element-ui的Tree树形组件有bug,会导致click事件丢失,从而无法添加数据
  • 本代码实现了增删的方法,修改也是拿到index,通过index去修改list中的数据
  • 索引没什么用,使用只会影响性能,但是没有办法,产品设计这样,只能单独递归渲染
  • 优化项很多,比如将计算索引的方法提取出来,在添加和删除阶段计算好,删除的时候通过父级只重新计算兄弟级的索引,这样可以节省性能
  • 我这里是只要发生变化,就会重新计算整个树的数据
  • 因为删除的时候只是删除了父级,所以会有脏数据,但是树的数据是干净的,提交到服务器的时候需要将树重新转换为1维数组
  • id是数据库的默认主键字段,在这里却是非常重要的父子级依赖关系,如果和服务端冲突需要替换为其他字段
  • 实际开发中这个逻辑由服务端实现比较好,但如果服务端不想写,那只能由前端完成
  • 数据格式应由服务端规定,免得生成的数据无法使用
  • 最好的结果就是每次增删改查一条数据都调用服务端接口
  • 前端实现时为了避免新增了半天的数据因为不小心刷新页面而丢失,可使用localStorange,在提交之后再将数据清除

下面HTML是示例代码,可将其中arrToTree方法替换,可以在浏览器直接打开运行

<!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset="UTF-8" />
    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
    <title>Document</title>
    <script src="https://cdn.jsdelivr.net/npm/vue@2/dist/vue.js"></script>
    <link
      rel="stylesheet"
      href="https://unpkg.com/element-ui/lib/theme-chalk/index.css"
    />
    <script src="https://unpkg.com/element-ui/lib/index.js"></script>
    <style>
      .lh {
        line-height: 200px;
      }
    </style>
  </head>
  <body>
    <div id="app">
      {{ message }}
      <p>use scoped slot</p>
      <el-button type="text" size="mini" @click="add">add</el-button>
      <el-tree
        :data="tree"
        show-checkbox
        node-key="id"
        default-expand-all
        :expand-on-click-node="false"
      >
        <span class="custom-tree-node lh" slot-scope="{ node, data }">
          <span class="lh">{{ node.label }}</span>
          <span class="lh">
            <el-button type="text" size="mini" @click="()=>append(data)">
              Append
            </el-button>
            <el-button type="text" size="mini" @click="()=>edit(data)">
              edit
            </el-button>
            <el-button type="text" size="mini" @click="remove(node, data)">
              Delete
            </el-button>
          </span>
        </span>
      </el-tree>
      <el-table
        :data="tree"
        style="width: 100%; margin-bottom: 20px"
        row-key="id"
        border
        default-expand-all
        :tree-props="{children: 'children', hasChildren: 'hasChildren'}"
      >
        <el-table-column prop="date" label="日期" sortable width="180">
        </el-table-column>
        <el-table-column prop="label" label="姓名" sortable width="180">
        </el-table-column>
        <el-table-column prop="number" label="地址"> </el-table-column>
        <el-table-column fixed="right" label="操作" width="100">
          <template slot-scope="scope">
            <el-button @click="append(scope.row)" type="text" size="small"
              >append</el-button
            >
            <el-button @click="remove(scope.row)" type="text" size="small">remove</el-button>
          </template>
        </el-table-column>
      </el-table>
    </div>
    <script>
      var app = new Vue({
        el: "#app",
        data: {
          message: "Hello Vue!",
          list: [],
          defaultProps: {
            children: "children",
            label: "label",
          },
        },
        computed: {
          tree() {
            let tree = this.arrToTree(this.list);
            console.log(tree);
            return tree;
          },
        },

        methods: {
          setChildrenIndex(params) {
            if (params.children && params.children.length) {
                params.children.forEach((item, index) => {
                item.number = params.number + "-" + (index + 1);
                item.label=item.label+item.number
                this.setChildrenIndex(item);
                return item;
              });
            }
            return params;
          },
          arrToTree(arr) {
            console.log("arr", arr);
            arr = JSON.parse(JSON.stringify(arr));
            let level = [];
            let count = 0;
            while (true) {
              console.log(count);
              let a = [];
              if (!level.length) {
                a = arr.filter((item) => 0 == item.pid);
                a = a.map((item, index) => {
                  item.number = "" + (index + 1);
                  item.label = item.label + item.number;
                  return item;
                });
              } else {
                a = arr.filter((item) => {
                  let arrs = level[count];
                  let c = arrs.find((ite) => {
                    return Boolean(ite.id == item.pid);
                  });
                  return !!c;
                });
                count++;
              }
              if (a.length > 0) {
                level.push(a);
              } else {
                break;
              }
            }
            // 反向循环挂载树
            let menu = [];
            for (let i = level.length - 1; i >= 0; i--) {
              if (level[i - 1]) {
                menu = level[i - 1].map((item) => {
                  item.children = level[i].filter((ite) => ite.pid == item.id);
                  return item;
                });
              } else {
                menu = level[0];
                break;
              }
            }
            // 正向循环渲染number 递归重新计算number
            menu = menu.map(item=>this.setChildrenIndex(item));
            return menu;
          },
          add() {
            let id = 1;
            if (this.list.length) {
              id = this.list[this.list.length - 1].id + 1;
            }
            this.list.push({ id, label: "张三", pid: 0 });
          },
          append(data) {
            let pid = data.id;
            let id = this.list[this.list.length - 1].id + 1;
            let item = { id, label: "张三", pid };
            console.log(item);
            this.list.push(item);
          },
          edit() {},
          remove(data) {
              let index = this.list.findIndex(item=>item.id==data.id)
              this.list.splice(index,1)
          },
        },
      });
    </script>
  </body>
</html>

二次优化

  • 优化思路,既然递归可以将索引循环出来,那么就可以将遍历子级的代码放在递归中实现,代码如下:
  • 去掉之前冗余的写法,将arrToTree方法改造一下
  • 既然map循环中只用到了第一层,那么只需要将第一层父级过滤出来即可
  • 最后将查找子级的方法放在递归方法中
  // 改造后的 arrToTree方法
       arrToTree(arr) {
            // 克隆数组避免引用引起原数组变化
            arr = JSON.parse(JSON.stringify(arr));
            // 去掉之前冗余的写法,将第一层父级过滤出来
            const menu = arr.filter((item) => item.pid == 0);
            // 正向循环渲染number 递归重新计算number
            return menu.map((item,index) => {
                item.number=`${index+1}`
                item.label = item.label+item.number
                return this.setChildrenIndex(item)
            });
          },
          // 改造后的 setChildrenIndex方法
          setChildrenIndex(params) {
            // 添加了寻找子级的方法
            let list = arr = JSON.parse(JSON.stringify(this.list));
            params.children = arr.filter((item) => item.pid == params.id);
            if (params.children && params.children.length) {
              params.children.forEach((item, index) => {
                item.number = params.number + "-" + (index + 1);
                item.label = item.label + item.number;
                this.setChildrenIndex(item);
                return item;
              });
            }
            return params;
          },

三次优化

  • 优化思路,既然递归可以实现将一维数组转树的需求,那么只需要将arrToTree和setChildrenIndex的方法合并成一个方法即可
  • 为了寻找它的子级,我们需要知道最先开始的父级id,所以递归方法中需要多一个参数pid
  • 添加索引参数,方便子级拼接
         arrToTree(list, pid = 0, str = "") {
            // 克隆数组避免引用引起原数组变化
            if (!pid) list = JSON.parse(JSON.stringify(list));
            let result = list.filter((item) => item.pid == pid);
            return result.map((item,index) => {
              item.number = `${str ? str + "-" : str}${index + 1}`;
              item.label = item.label + item.number
              item.children = this.arrToTree(list, item.id, item.number);
              return item
            });
          },

其他解法

  • 朋友写的,据说减少了遍历次数
arrToTree(arr,pid=null,number=null){
  const res = [];
  let count = 1;
  arr.forEach((item)=>{
    if(item.pid == pid){
       if(number=='-'){
          item.number = count
       }else{
          item.number = number + '-'  + number
      }
    }
    // 随着filter的执行,数组长度会越来越小,这与全量循环相比,减少了很多循环次数
    const children = arrToTree(arr.filter(v=>v.pid!==pid),item.id,item.number)
    if(children.length){
      res.push({...item,children})
    }else{
      res.push({...item})
    }
  })
  return res
}

优秀的程序员不在于他写了多少行代码,而在于用简单的方式去解决问题,编程是一种艺术,好的代码根据优化和重构,可以将复杂的问题简单化

相关文章