将含有父ID的列表转成树

我们知道数据库一般是以一个列表(id,pid)的形式保存树的。如何提取这棵树呢？最简单的方法就是根据pid循环查表。但是毫无疑问，这会产生巨大的数据库查询开销。

那么一般建议的方法是一次性将全部相关数据全查出来，但是这就涉及到一个问题，如何快速的构建一棵树。

我曾经一直以为，这是一个复杂的操作，至少需要一个递归，时间复杂度不会是O(n)。

前段时间，一个工作上的需求，需要解决这个问题。我仔细想了想，发现完全可以通过单层循环解决这个问题，实现如下：

 function list2Tree($listItem, $idx = 'id', $pIdx = 'pid', $childKey= 'list'){
     $map = array();
     $pMap = array();
 
     foreach($listItem as $item){
         $id = $item[$idx];
         $pid = $item[$pIdx];
         $map[$id] = &$item;
         unset($item);
     }       
 
     foreach($map as $id => &$item){
         $pid = $item[$pIdx];
         empty($item[$childKey]) && $item[$childKey] = array();
 
         if(! isset($map[$pid])){
             $pMap[$id] = &$item;
         }
         else{
             $pItem= &$map[$pid];
             $pItem[$childKey][] = &$item;
         }   
 
         unset($item, $pItem);
     }
 
     return array_shift($pMap);
 }

测试一下：

 // 路径方便识别父子关系
 $json = <<<JSON
 [
     {
         "id": 2,
         "pid": 1,
         "path": "/se"
     },
     {
         "id": 3,
         "pid": 2,
         "path": "/se/4901"
     },
     {
         "id": 4,
         "pid": 5,
         "path": "/se/4901/mask/query"
     },
         {
         "id": 5,
         "pid": 3,
         "path": "/se/4901/mask"
     },
     {
         "id": 6,
         "pid": 2,
         "path": "/se/4902"
     },
     {
         "id": 7,
         "pid": 6,
         "path": "/se/4902/mask"
     }
 ]
 JSON;
 
 $list = json_decode($json, true);
 
 var_dump(list2Tree($list));

结果：

array(4) {
  ["id"]=>
  int(2)
  ["pid"]=>
  int(1)
  ["path"]=>
  string(3) "/se"
  ["list"]=>
  array(2) {
    [0]=>
    array(4) {
      ["id"]=>
      int(3)
      ["pid"]=>
      int(2)
      ["path"]=>
      string(8) "/se/4901"
      ["list"]=>
      array(1) {
        [0]=>
        array(4) {
          ["id"]=>
          int(5)
          ["pid"]=>
          int(3)
          ["path"]=>
          string(13) "/se/4901/mask"
          ["list"]=>
          array(0) {
          }
        }
      }
    }
    [1]=>
    array(4) {
      ["id"]=>
      int(6)
      ["pid"]=>
      int(2)
      ["path"]=>
      string(8) "/se/4902"
      ["list"]=>
      array(1) {
        [0]=>
        array(4) {
          ["id"]=>
          int(7)
          ["pid"]=>
          int(6)
          ["path"]=>
          string(13) "/se/4902/mask"
          ["list"]=>
          array(0) {
          }
        }
      }
    }
  }
}

成功把列表转成了树