造树
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
}