在最近的项目中，我需要从数据库中挑选出带有层级关系的数据，并根据记录的parent_id构建数据之间的父子关系，最终形成一个树状结构的父子孙关系结构，并用数组的形式保存起来。经过搜索和研究，最终找到了下方的算法代码，仅做参考。

首先，查询得到的数据以数组的形式返回：

array(
  array('id' => 5,'parent_id' => 0,...),
  array('id' => 9,'parent_id' => 0,...),
  array('id' => 12,'parent_id' => 5,...),
  array('id' => 19,'parent_id' => 5,...),
  array('id' => 21,'parent_id' => 0,...),
  array('id' => 23,'parent_id' => 12,...),
  array('id' => 24,'parent_id' => 12,...),
  array('id' => 26,'parent_id' => 23,...),
  array('id' => 29,'parent_id' => 9,...),
  array('id' => 31,'parent_id' => 12,...),
  ...
)

通过观察，我们可以发现顶级数组是由包含id,parent_id的数组组成的，这些数组之间存在着一定的父子关系，而且仅以parent_id的形式表现出来。现在，我们实际上想要得到下面的结果：

array(
  array(
    'id' => 5,
    'children' => array(
      array(
        'id' => 12,
        'children' => array(
          array(
            'id' => 23,
            'children' => array(
              array('id' => 26)
            )
          ),
          array('id' => 24),
          array('id' => 31)
        )
      ),
      array('id' => 19)
    )
  ),
  array(
    'id' => 9,
    'children' => array(
      array('id' => 29)
    )
  ),
  array('id' => 21),
  ...
)

也就是通过数组的形式，用一个children元素来包含该元素的子元素，并且层级分明。

那么如何来快速实现呢？我们通过下面一个函数来实现。

function array_tree($array) {
  $result = array();
  $tmp = array();
  foreach($array as $item) {
    if($item['parent_id'] == 0) {
      $i = count($result);
      $result[$i] = $item;
      $id = $item['id'];
      $tmp[$id] = &$result[$i];
    }
    else {
      $id = $item['id'];
      $parent_id = $item['parent_id'];
      $parent = $tmp[$parent_id];
      $i = count($parent['children']);
      $tmp[$parent_id]['children'][$i] = $item;
      $tmp[$id] = &$tmp[$parent_id]['children'][$i];
    }
  }
  return $result;
}

上面我用红色标注了&符号，具体是什么意思，及其原理又是怎么实现的呢？

&引用是PHP中非常特殊的一种变量引用方式。我们不过多深入的去讲解，简单的讲，就是：

当$b = &$a时，$b和$a同时引用同一个内容（指针指向同一块内存），无论$a或$b谁发生变化，这个内容都会发生变化，进而呈现为$a和$b保持同步的变化。

要解除这种引用，只有先将引用释放。可是并没有一种运算可以达到释放引用的目的，所以我们可以通过unset($a)的方式，也可以通过重新引用另外一个变量来释放原来的引用（但是又和其他变量引用，比如$b = &$c）。

这时，我们再来分析上面的实现函数。

$tmp[$id] = &$result[$i];

这一句保证了当无论$tmp[$id]还是$result[$i]发生变化，都会让另外一个值同时发生变化。而

$tmp[$parent_id]['children'][$i] = $item;

实际上导致$result[$parent_id]发生了变化。

用我们上面的数据进行演示验证：

[0] => array('id' => 5,'parent_id' => 0,...)

$result[0] = array('id' => 5,'parent_id' => 0,...)

$tmp[5] = &$result[0]  // 第一个&出现了

$tmp[5] = array('id' => 5,'parent_id' => 0,...)  // 内部结果

而当执行往下执行，碰到array('id' => 12,'parent_id' => 5,...)时，发生了：

[2]=> array('id' => 12,'parent_id' => 5,...)

$tmp[5]['children'][0] = array('id' => 12,'parent_id' => 5,...) // ①

由于&引用的关系，所以实际上

$result[0]['children'][0] = array('id' => 12,'parent_id' => 5,...) // 内部结果

这个时候$temp[5]是一个已经包含children子元素的二维数组，如果往下执行遇到array('id' => 23,'parent_id' => 12,...)时，你就会发现

[5]=> array('id' => 23,'parent_id' => 12,...)

$tmp[12]['children'][0] = array('id' => 23,'parent_id' => 12,...)

$result[?]['children'][0] = array('id' => 23,'parent_id' => 12,...) // 我们以为会发生的结果，其实并没有发生

因为没有找到一个$tmp[12] = & $result[?]，也就是说没有一个与$tmp[12]对应的引用，$tmp[12]的变化不会引起其他任何值的变化，$tmp[12]这个家伙会在返回值中丢失（因为返回值是$result）。

而$tmp[$id] = &$tmp[$parent_id]['children'][$i];这一句起到了关键性作用。加上这一句之后，我们再来跑一遍array('id' => 23,'parent_id' => 12,...)这个元素。

[2]=> array('id' => 12,'parent_id' => 5,...)

$tmp[5]['children'][0] = array('id' => 12,'parent_id' => 5,...)  // 重新从①开始演示

$result[0]['children'][0] = array('id' => 12,'parent_id' => 5,...) // 内部结果

$tmp[12] = &$tmp[5]['children'][0]  // 第二个&出现了

[5]=> array('id' => 23,'parent_id' => 12,...)

$tmp[12]['children'][0] = array('id' => 23,'parent_id' => 12,...)

由于第二个&引用的原因，实际上发生了：

$tmp[5]['children'][0]['children'][0] = array('id' => 23,'parent_id' => 12,...) // 内部结果

$result[0]['children'][0]['children'][0] = array('id' => 23,'parent_id' => 12,...) // 内部结果

这个时候你可能已经看出了端倪。父子关系变成了 5 > 12 > 23，而这组关系，全部存储在了id=5的这个顶级元素中，以多重的children元素实现了父子孙结构。

如此循环往复，你就会发现，无论你的父子关系多深，这个函数都能准确的构建对应的父子关系，最终形成树形结构存储在$result中，将其返回。

该算法有一个问题，就是要求原始数组中，parent_id必须出现在该元素之前，否则将导致该元素无法加入到某个元素的children中去。

2015-11-16 6168