1,
/**
* 将一个平面的二维数组按照指定的字段转换为树状结构
*
* 用法:
* @code php
* $rows = array(
* array('id' => 1, 'value' => '1-1', 'parent' => 0),
* array('id' => 2, 'value' => '2-1', 'parent' => 0),
* array('id' => 3, 'value' => '3-1', 'parent' => 0),
*
* array('id' => 7, 'value' => '2-1-1', 'parent' => 2),
* array('id' => 8, 'value' => '2-1-2', 'parent' => 2),
* array('id' => 9, 'value' => '3-1-1', 'parent' => 3),
* array('id' => 10, 'value' => '3-1-1-1', 'parent' => 9),
* );
*
* $tree = ArrayHelper::tree($rows, 'id', 'parent', 'nodes');
*
* dump($tree);
* // 输出结果为:
* // array(
* // array('id' => 1, ..., 'nodes' => array()),
* // array('id' => 2, ..., 'nodes' => array(
* // array(..., 'parent' => 2, 'nodes' => array()),
* // array(..., 'parent' => 2, 'nodes' => array()),
* // ),
* // array('id' => 3, ..., 'nodes' => array(
* // array('id' => 9, ..., 'parent' => 3, 'nodes' => array(
* // array(..., , 'parent' => 9, 'nodes' => array(),
* // ),
* // ),
* // )
* @endcode
*
* 如果要获得任意节点为根的子树,可以使用 $refs 参数:
* @code php
* $refs = null;
* $tree = ArrayHelper::tree($rows, 'id', 'parent', 'nodes', $refs);
*
* // 输出 id 为 3 的节点及其所有子节点
* $id = 3;
* dump($refs[$id]);
* @endcode
*
* @param array $arr 数据源
* @param string $keyNodeId 节点ID字段名
* @param string $keyParentId 节点父ID字段名
* @param string $keyChildrens 保存子节点的字段名
* @param boolean $refs 是否在返回结果中包含节点引用
*
* return array 树形结构的数组
*/
static function toTree($arr, $keyNodeId, $keyParentId = 'parent_id', $keyChildrens = 'childrens', & $refs = NULL)
{
$refs = array();
foreach ($arr as $offset => $row)
{
$arr[$offset][$keyChildrens] = array();
$refs[$row[$keyNodeId]] =& $arr[$offset];
} $tree = array();
foreach ($arr as $offset => $row)
{
$parentId = $row[$keyParentId];
if ($parentId)
{
if (!isset($refs[$parentId]))
{
$tree[] =& $arr[$offset];
continue;
}
$parent =& $refs[$parentId];
$parent[$keyChildrens][] =& $arr[$offset];
}
else
{
$tree[] =& $arr[$offset];
}
}
return $tree;
} /**
* 将树形数组展开为平面的数组
*
* 这个方法是 tree() 方法的逆向操作。
*
* @param array $tree 树形数组
* @param string $keyChildrens 包含子节点的键名
*
* @return array 展开后的数组
*/
static function treeToArray($tree, $keyChildrens = 'childrens')
{
$ret = array();
if (isset($tree[$keyChildrens]) && is_array($tree[$keyChildrens]))
{
foreach ($tree[$keyChildrens] as $child)
{
$ret = array_merge($ret, self::treeToArray($child, $keyChildrens));
}
unset($node[$keyChildrens]);
$ret[] = $tree;
}
else
{
$ret[] = $tree;
}
return $ret;
}