将返回的数据集转换成树

原创: 2016-06-07 11:45:44 984浏览

/**

 * 将返回的数据集转换成树

 * @param  array   $list  数据集

 * @param  string  $pk    主键

 * @param  string  $pid   父节点名称

 * @param  string  $child 子节点名称

 * @param  integer $root  根节点ID

 * @return array          转换后的树

 */

function list_to_tree($list, $pk = 'id', $pid = 'pid', $child = '_child', $root=0) {

    $tree = array();// 创建Tree

    if(is_array($list)) {

        // 创建基于主键的数组引用

        $refer = array();

        foreach ($list as $key => $data) {

            $refer[$data[$pk]] =& $list[$key];

        }

        

        foreach ($list as $key => $data) {

            // 判断是否存在parent

            $parentId = $data[$pid];

            if ($root == $parentId) {

                $tree[$data[$pk]] =& $list[$key];

            }else{

                if (isset($refer[$parentId])) {

                    $parent =& $refer[$parentId];

                    $parent[$child][] =& $list[$key];

                }

            }

        }

    }

    return $tree;

}