树的常用方法

294 阅读1分钟

造树

const data = [
{ id: 1, name: "1", pid: 0,flag:1 },
{ id: 2, name: "2", pid: 1,flag:1  },
{ id: 3, name: "3", pid: 1 },
{ id: 4, name: "4", pid: 2,flag:1  },
{ id: 5, name: "5", pid: 0 },
{ id: 6, name: "6", pid: 5,flag:1  },
{ id: 7, name: "7", pid: 6,flag:1  },
{ id: 8, name: "8", pid: 6 },
];

会修改数据源
// 
  toTree(data) {
    // data = JSON.parse(JSON.stringify(data)) // 此步骤不影响数据源,data需要额外赋值进行处理
  
        // 删除 所有 children,以防止多次调用
        data.forEach(function (item) {
            delete item.children;
        });
 
        // 将数据存储为 以 id 为 KEY 的 map 索引数据列
        var map = {};
        data.forEach(function (item) {
            map[item.id] = item;
        });
//        console.log(map);
        var val = [];
        data.forEach(function (item) {
            // 以当前遍历项,的pid,去map对象中找到索引的id
            var parent = map[item.pid];
            // 好绕啊,如果找到索引,那么说明此项不在顶级当中,那么需要把此项添加到,他对应的父级中
            if (parent) {
                (parent.children || ( parent.children = [] )).push(item);
            } else {
                //如果没有在map中找到对应的索引ID,那么直接把 当前的item添加到 val结果集中,作为顶级
                val.push(item);
            }
        });
        return val;
    }

取出树中flag为1的对象的id

1.递归

const arr =[]
getFlagData(data: any) {
  for (let a of data) {
    if (a && a.child) {
      this.getFlagData(a.child);
    } 
    if (a && a.flag) {
      arr.push(a.id);
    }
  }
  return arr
}

2.遍历

getFlagData(node){
  if (!node) {
    return;
  }
  var stack = [];
  stack.push(...node);
  var tmpNode;
  var result = []
  while (stack.length > 0) {
    tmpNode = stack.pop();
    if (tmpNode.child && tmpNode.child.length > 0) {
      var i = tmpNode.child.length - 1;
      for (i = tmpNode.child.length - 1; i >= 0; i--) {
        stack.push(tmpNode.child[i]);
      }
    }
    if(tmpNode.flag){
      result.push(tmpNode.id)
    }
  }
  return result
}