递归

// 角色数据
const roleData = [
  {
    id: 1, //角色id
    name: "网络服务部", // 角色名称
    parent_id: 0, // 父类没有 parent_id
  },
  {
    id: 2, //角色id
    name: "前端开发组", // 角色名称
    parent_id: 1, // 父类没有 parent_id
  },
  {
    id: 3, // 角色id
    name: "前端开发小白", // 角色名称
    parent_id: 2, // 父类没有 parent_id
  },
  {
    id: 4, //角色id
    name: "SASS平台组", // 角色名称
    parent_id: 1, // 父类没有 parent_id
  },
  {
    id: 5, //角色id
    name: "SASS开发小黄", // 角色名称
    parent_id: 4, // 父类没有 parent_id
  },
  {
    id: 6, //角色id
    name: "安卓平台部", // 角色名称
    parent_id: 0, // 父类没有 parent_id
  },
  {
    id: 7, //角色id
    name: "安卓产品部", // 角色名称
    parent_id: 6, // 父类没有 parent_id
  },
  {
    id: 8, //角色id
    name: "产品开发1", // 角色名称
    parent_id: 7, // 父类没有 parent_id
  },
  {
    id: 9, //角色id
    name: "产品开发2", // 角色名称
    parent_id: 7, // 父类没有 parent_id
  },
];

// 递归
function listToTree(data: RoleType[]): RoleType[] {
  const map = new Map();
  const tree: RoleType[] = [];
  data.forEach((item) => {
    map.set(item.id, { ...item, children: [] });
  });

  data.forEach((item) => {
    const parent = map.get(item.parent_id);
    if (parent) {
      parent.children.push(map.get(item.id));
    } else {
      tree.push(map.get(item.id));
    }
  });
  return tree;
}

 

解递归

function treeToList(tree) {
  const result = [];

  function recursive(node) {
    result.push(node);
    if (node.children && node.children.length > 0) {
      node.children.forEach((child) => {
        recursive(child);
      });
    }
  }

  tree.forEach((node) => {
    recursive(node);
  });

  return result;
}