JS 樹形結構與數組結構相互轉換、在樹形結構中查找對象

總是有很多需求是關於處理樹形結構的，所以不得不總結幾個常見操作的寫法。¯\_(ツ)_/¯

首先假設有一個樹形結構數據如下

var tree=[
  {
    'id': '1',
    'name': '教學素材管理',
    'children':[
      {
        'id': '101',
        'name': '教學素材',
        'children':[
          {
            'id': '10101',
            'name': '修改',
          },
          {
            'id': '10102',
            'name': '添加',
          }
        ]
      },  {
        'id': '102',
        'name': '測試試題',
      },
      {
        'id': '103',
        'name': '問題任務',
      }
    ]
  }, {
    'id': '2',
    'name': '基礎數據管理',
    'children':[
      {
        'id': '201',
        'name': '專業設置',
      },
      {
        'id': '202',
        'name': '專業管理',
      }
    ]
  }
]

1、如何在tree中找到id=10102的對象？

思路一：深度遍歷，從頂點開始，當前節點有子節點則遍歷當前節點的子節點（遞歸）。

function deepQuery(tree,id) {
    var isGet = false;
    var retNode = null;
    function deepSearch(tree,id){
        for(var i = 0; i<tree.length; i++) {
            if(tree[i].children && tree[i].children.length>0) {
                deepSearch(tree[i].children,id);
            }
            if(id === tree[i].id || isGet) {
                isGet||(retNode = tree[i]);
                isGet = true;
                break;
            }
        }
    }
    deepSearch(tree,id);
    return retNode;
}

var getNode = deepQuery(tree,10102);
console.log(getNode)

思路二：廣度遍歷，遍歷根節點的所有子節點，再從第一個子節點開始依次遍歷。

function breadthQuery(tree, id) {
    var stark = [];
    stark = stark.concat(tree);
    while(stark.length) {
        var temp = stark.shift();
        if(temp.children) {
            stark = stark.concat(temp.children);
        }
        if(temp.id === id) {
            return temp;
        }
    }
}

var getNode=breadthQuery(tree,10102);
console.log(getNode);

2、如何將樹形結構轉換為有父子關系屬性的數組結構？

思路一：初始化一個空數組，從tree的頂端開始遍歷，當前節點有子節點時，一邊繼續遍歷子節點，一邊在當前節點上刪除子節點，將當前節點push到空數組。

function flatten1(tree) {
    var arr = [];
    function spread(tree,pid) {
        for (var i=0; i < tree.length; i++ ) {
            item = tree[i]
            let {id,name}=item;
            arr.push({id,name,pid})
            if (item.children) {
                spread(item.children,item.id)
                delete item.children
            }
        }
    }
    spread(tree,0)
    return arr;
}
var newArr = flatten1(tree);

思路二：

function flatten2 (data,pid) {
  return data.reduce((arr, {id, name, children = []}) =>
    arr.concat([{id, name,pid}], flatten2(children,id)), [])
} 
var newArr = flatten2(tree,0);

結果：

3、如何將數組結構轉換為樹形結構？

下面是偶然看到一位大佬很秀的寫法（讀書人的事怎么能叫抄呢ヽ(ﾟ∀ﾟ)ﾒ(ﾟ∀ﾟ)ﾉ 原文鏈接 ）

function treeData(data){   
    let cloneData = JSON.parse(JSON.stringify(data))
    return cloneData.filter(parent=>{
        let branchArr = cloneData.filter(child => parent['id'] == child['pid']);
        branchArr.length>0 ? parent['children'] = branchArr : '';
        return parent['pid'] == 0 ;
    })
}

var newTree = treeData(newArr)

用之前測試生成的數組試一下

 結果如上，完美的生成了最初的樹形結構。