PHP递归实现数组转树形结构方法

1. 引言：树形结构数据处理的挑战

在Web开发中，处理具有层级关系的数据是一项常见任务，例如网站导航菜单、商品分类、评论回复、组织架构等。这类数据通常以扁平化的形式存储在数据库中，每条记录包含一个ID和一个指向其父级记录的ID（parentid）。然而，在前端展示或进行某些业务逻辑处理时，我们往往需要将这种扁平数据转换为嵌套的树形结构。

例如，我们可能有如下的扁平化数据：

$indexes = [
    ['id' => 1, 'parentid' => 0, 'route' => 'root', 'title' => 'root'],
    ['id' => 2, 'parentid' => 1, 'route' => 'parent', 'title' => 'parent'],
    ['id' => 3, 'parentid' => 2, 'route' => 'child', 'title' => 'child']
];

我们期望将其转换为如下的嵌套结构，其中子元素通过 pages 键包含：

$index = [
  [
    'id' => 1,
    'pages' => [
       [
         'id' => 2,
         'pages' => [
          [
            'id' => 3
          ]
        ]
      ]
    ]
  ]
];

2. 核心概念：递归

递归是一种函数或过程调用自身的编程技术。在处理树形结构数据时，递归表现出其天然的优势。构建树形结构的过程可以被分解为：找到当前父节点的所有直接子节点，然后对每个子节点重复相同的过程（即找到它们的子节点），直到没有更多的子节点为止。这正是递归思想的完美应用场景。

3. 构建树形结构的递归函数

我们将创建一个名为 buildSubs 的函数，它接收两个参数：完整的扁平化数据数组 $elms 和当前要查找的父ID $parentId。

3.1 函数逻辑解析

1. 初始化分支数组： 在函数内部，首先初始化一个空数组 $branch，用于存放当前 $parentId 下的所有直接子元素。
2. 遍历所有元素： 遍历传入的 $elms 数组中的每一个元素 $elm。
3. 识别直接子元素： 检查当前元素 $elm 的 parentid 是否等于 $parentId。如果相等，则说明 $elm 是 $parentId 的一个直接子元素。
4. 递归构建子树： 如果 $elm 是直接子元素，则递归调用 buildSubs 函数，传入完整的 $elms 数组和当前子元素的 id ($elm['id']) 作为新的 $parentId。这将返回当前子元素的所有后代组成的子树。
5. 挂载子树： 如果递归调用返回了子树（即 $children 不为空），则将其赋值给当前元素 $elm 的 pages 键。注意：这里是常见的错误点，必须是 $elm['pages'] = $children; 而不是 $elms['pages'] = $children;，因为我们是要修改当前正在处理的单个元素，而不是整个原始数组。
6. 添加到分支： 将处理好的 $elm（可能已经包含了其子树）添加到 $branch 数组中。
7. 返回分支： 循环结束后，返回 $branch 数组，它包含了 $parentId 下的所有直接子元素及其完整的子树。

3.2 初始调用与根节点处理

为了构建完整的树形结构，我们需要从根节点开始。在我们的示例数据中，根节点的 parentid 是 0。因此，在第一次调用 buildSubs 函数时，$parentId 应该设置为 0。

4. 完整示例代码

 $elm) { // 遍历所有元素
        if ($elm['parentid'] == $parentId) { // 如果当前元素的parentid匹配目标parentId
            // 递归调用自身，查找当前元素的子元素
            $children = buildSubs($elms, $elm['id']);

            // 如果存在子元素，则将其添加到当前元素的 'pages' 键中
            if (!empty($children)) {
                $elm['pages'] = $children; // 核心修正：修改 $elm 而非 $elms
            }

            // 将处理好的元素（可能已包含子树）添加到当前分支
            $branch[] = $elm;

            // 优化：从原数组中移除已处理的元素，减少后续遍历的范围 (可选，但对于大型数据集有性能优势)
            // unset($elms[$key]); // 注意：如果使用此行，递归调用时需要传递引用或重新考虑逻辑
        }
    }
    return $branch;
}

// 原始扁平化数据
$indexes = [
    ['id' => 1, 'parentid' => 0, 'route' => 'root', 'title' => 'root'],
    ['id' => 2, 'parentid' => 1, 'route' => 'parent', 'title' => 'parent'],
    ['id' => 3, 'parentid' => 2, 'route' => 'child', 'title' => 'child']
];

// 从根节点（parentid为0）开始构建完整的树
$tree = buildSubs($indexes, 0);

// 输出结果
echo '
';
var_dump($tree);
echo '
';

?>

5. 运行结果展示

执行上述代码后，var_dump($tree) 将输出以下结果，这正是我们期望的树形结构：

Array
(
    [0] => Array
        (
            [id] => 1
            [parentid] => 0
            [route] => root
            [title] => root
            [pages] => Array
                (
                    [0] => Array
                        (
                            [id] => 2
                            [parentid] => 1
                            [route] => parent
                            [title] => parent
                            [pages] => Array
                                (
                                    [0] => Array
                                        (
                                            [id] => 3
                                            [parentid] => 2
                                            [route] => child
                                            [title] => child
                                        )
                                )
                        )
                )
        )
)

6. 注意事项与最佳实践

• 性能考量： 对于非常庞大或层级非常深的数据集，递归可能会导致性能问题（如栈溢出或多次重复遍历）。在这种情况下，可以考虑使用迭代方法（如基于引用的方法）或在数据库层面进行优化查询。
• 内存消耗： 深度嵌套的数组结构会占用更多内存。确保您的服务器配置能够处理预期的数据量。
• 灵活性： 为了使函数更通用，可以将其参数化，允许用户指定ID键名、父ID键名和子数组键名（例如：'id_key', 'parent_id_key', 'children_key'）。
• 错误处理：
  • 循环引用： 如果数据中存在A是B的父，B是A的父的循环引用，递归函数将陷入无限循环。在实际应用中，应避免此类数据结构或在递归中加入深度限制或已访问节点记录。
  • 孤儿节点： 如果某个元素的 parentid 指向一个不存在的ID，它将不会被包含在任何分支中，除非您有特定的逻辑来处理这些“孤儿”节点。
• 优化遍历： 在 foreach 循环内部，如果找到并处理了一个元素，可以考虑从原始 $elms 数组中将其移除（使用 unset($elms[$key])）。这样可以减少后续迭代的元素数量，从而提高效率。但请注意，如果这样做，递归调用时需要确保 $elms 数组的副本或引用传递方式是正确的。上述示例代码为了简洁和避免潜在的引用复杂性，并未采用此优化。

7. 总结

通过本教程，我们学习了如何利用PHP的递归功能将扁平化的父子关系数据转换为易于处理和展示的树形结构。理解递归的核心逻辑、正确处理当前元素的修改以及从正确的根节点开始构建是实现这一转换的关键。虽然递归在处理树形数据时非常优雅，但在面对大规模数据时也需要考虑其性能和内存影响，并根据实际需求选择最合适的实现方案。