递归函数遍历多维数组输出叶子节点信息的实现与问询

我编写了一个递归函数，可遍历至多维数组的最末层级并打印节点的层级、键与对应值，函数代码如下：

function recursive($array, $level = 1) { 
    foreach($array as $key => $value){ 
        if(is_array($value)){
            recursive($value, $level + 1); 
        } else{
            echo 'LEVEL : '. $level . " ==> " . $key . ": " . $value, '<br>'; 
        } 
    } 
}

该函数的输出示例如下：

LEVEL : 2 ==> value: 10
LEVEL : 4 ==> key: A
LEVEL : 5 ==> value: 15
LEVEL : 7 ==> key: B
LEVEL : 8 ==> value: 45
LEVEL : 4 ==&...

请问该实现是否存在可优化之处？

当然有不少可以优化的地方，我从实用性、灵活性和性能几个角度给你梳理下：

• 分离逻辑与输出，提升复用性
  现在你的函数直接用echo输出结果，这会让函数的职责太单一——只能用来打印，没法把遍历结果拿去做其他处理（比如存入数据库、生成数组报表）。建议把输出逻辑抽离出来，让函数返回遍历后的结果数组，或者支持传入回调函数来处理每个节点：

  function recursive($array, $level = 1, callable $handler = null) {
      $result = [];
      foreach($array as $key => $value){
          if(is_iterable($value)){
              $result = array_merge($result, recursive($value, $level + 1, $handler));
          } else {
              $node = ['level' => $level, 'key' => $key, 'value' => $value];
              if($handler){
                  $handler($node);
              }
              $result[] = $node;
          }
      }
      return $result;
  }
  
  // 使用示例：打印输出
  recursive($yourArray, 1, function($node){
      echo "LEVEL : {$node['level']} ==> {$node['key']}: {$node['value']}<br>";
  });
  
  // 使用示例：获取所有节点数组
  $allNodes = recursive($yourArray);
  

  这样函数既可以用来打印，也能获取数据做后续操作，灵活性高很多。

• 兼容更多可遍历类型
  目前用is_array($value)判断只能处理普通数组，没法兼容ArrayObject这类实现了数组接口的对象。如果你的业务场景可能涉及这类对象，建议改用is_iterable($value)来判断，这样能覆盖所有可遍历的类型。

• 添加参数校验，避免非法输入
  最好在函数开头对传入的$array参数做校验，确保它是可遍历的，避免传入非遍历类型导致脚本报错：

  function recursive($array, $level = 1, callable $handler = null) {
      if(!is_iterable($array)){
          throw new InvalidArgumentException('First argument must be an iterable type');
      }
      // 后续遍历逻辑...
  }
  

• 替换递归为迭代，避免栈溢出
  如果你的多维数组层级非常深（比如超过1000层），递归调用会触发PHP的栈溢出错误。这种情况下可以改成迭代遍历的方式，用栈来模拟递归过程：

  function iterativeTraverse($array) {
      $stack = [[$array, 1]];
      $result = [];
      while(!empty($stack)){
          list($currentArray, $level) = array_pop($stack);
          foreach($currentArray as $key => $value){
              if(is_iterable($value)){
                  array_push($stack, [$value, $level + 1]);
              } else {
                  $node = ['level' => $level, 'key' => $key, 'value' => $value];
                  $result[] = $node;
                  // 也可以在这里添加回调处理逻辑
              }
          }
      }
      return $result;
  }
  

  迭代方式不会受递归深度限制，适合处理超深层级的数组。

• 让输出格式更灵活
  现在的输出格式是硬编码的（比如固定的<br>换行、固定的字符串模板），可以把这些做成可选参数，让用户自定义：

  function recursive($array, $level = 1, $lineBreak = '<br>', $format = "LEVEL : %d ==> %s: %s") {
      foreach($array as $key => $value){
          if(is_iterable($value)){
              recursive($value, $level + 1, $lineBreak, $format);
          } else {
              echo sprintf($format, $level, $key, $value) . $lineBreak;
          }
      }
  }
  

  这样用户可以根据需要改成\n换行，或者调整输出的字符串样式，适配不同的场景。

内容的提问来源于stack exchange，提问作者user9653591