天天看點

将傳回的資料集轉換成樹

/**
 * 将傳回的資料集轉換成樹
 * @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;
}