递归重组节点信息为多维数组

在一些项目中我们会有一些这样的需求，我们会将一张表中的数据根据父节点给它生成多位数组，下面我将分享两种方式：

 

/**
 * 递归重组节点信息为多维数组
 *	$node 节点数组
 *	$pid  父级ID
 */
function node_merge($node,$pid=0){
	$arr = array();
	foreach ($node as $v) {
		if ($v['pid']==$pid) {
			$v['child']=node_merge($node,$v['id']);
			$arr[]=$v;
		}
	}

	return $arr;
}

/**
 * 将返回的数据集转换成树
 * @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;
}

测试数据：

调用方法之前

array(99) {
  [0] => array(5) {
    ["id"] => string(1) "1"
    ["name"] => string(5) "Index"
    ["title"] => string(6) "首页"
    ["pid"] => string(1) "0"
    ["child"] => NULL
  }
  [1] => array(4) {
    ["id"] => string(1) "2"
    ["name"] => string(7) "addUser"
    ["title"] => string(12) "增加用户"
    ["pid"] => string(1) "7"
  }
  [2] => array(4) {
    ["id"] => string(1) "3"
    ["name"] => string(8) "changPwd"
    ["title"] => string(12) "修改密码"
    ["pid"] => string(2) "33"
  }
  [3] => array(4) {
    ["id"] => string(1) "4"
    ["name"] => string(8) "editUser"
    ["title"] => string(12) "编辑用户"
    ["pid"] => string(1) "7"
  }
  [4] => array(4) {
    ["id"] => string(1) "5"
    ["name"] => string(12) "index_layout"
    ["title"] => string(12) "首页格局"
    ["pid"] => string(1) "1"
  }
  [5] => array(4) {
    ["id"] => string(1) "6"
    ["name"] => string(5) "index"
    ["title"] => string(12) "首页内容"
    ["pid"] => string(1) "1"
  }
  [6] => array(4) {
    ["id"] => string(1) "7"
    ["name"] => string(11) "userManager"
    ["title"] => string(12) "用户管理"
    ["pid"] => string(2) "33"
  }
   ...
}

调用方法之后：

array(4) {
  [0] => array(5) {
    ["id"] => string(1) "1"
    ["name"] => string(5) "Index"
    ["title"] => string(6) "首页"
    ["pid"] => string(1) "0"
    ["child"] => array(5) {
      [0] => array(5) {
        ["id"] => string(1) "5"
        ["name"] => string(12) "index_layout"
        ["title"] => string(12) "首页格局"
        ["pid"] => string(1) "1"
        ["child"] => array(0) {
        }
      }
      [1] => array(5) {
        ["id"] => string(1) "6"
        ["name"] => string(5) "index"
        ["title"] => string(12) "首页内容"
        ["pid"] => string(1) "1"
        ["child"] => array(0) {
        }
      }
      [2] => array(5) {
        ["id"] => string(2) "31"
        ["name"] => string(4) "user"
        ["title"] => string(12) "角色管理"
        ["pid"] => string(1) "1"
        ["child"] => array(1) {
          [0] => array(5) {
            ["id"] => string(2) "32"
            ["name"] => string(11) "roleManager"
            ["title"] => string(12) "角色管理"
            ["pid"] => string(2) "31"
            ["child"] => array(0) {
            }
          }
        }

    ...
}