将$tree 中的平面结构转换为层次结构的另一种更简化的方法。公开它只需要一个临时数组:
// add children to parents
$flat = array(); # temporary array
foreach ($tree as $name => $parent)
{
$flat[$name]['name'] = $name; # self
if (NULL === $parent)
{
# no parent, is root element, assign it to $tree
$tree = &$flat[$name];
}
else
{
# has parent, add self as child
$flat[$parent]['children'][] = &$flat[$name];
}
}
unset($flat);
这就是将层次结构放入多维数组的全部内容:
Array
(
[children] => Array
(
[0] => Array
(
[children] => Array
(
[0] => Array
(
[name] => H
)
[1] => Array
(
[name] => F
)
)
[name] => G
)
[1] => Array
(
[name] => E
[children] => Array
(
[0] => Array
(
[name] => A
)
[1] => Array
(
[children] => Array
(
[0] => Array
(
[name] => B
)
)
[name] => C
)
)
)
)
[name] => D
)
如果您想避免递归(可能是大型结构的负担),输出就不那么简单了。
我一直想解决输出数组的 UL/LI “困境”。困境是,每个项目不知道孩子是否会跟进或需要关闭多少前面的元素。在另一个答案中,我已经通过使用RecursiveIteratorIterator 并寻找getDepth() 和我自己编写的Iterator 提供的其他元信息解决了这个问题:Getting nested set model into a <ul> but hiding “closed” subtrees。 answer 也表明,使用迭代器时您非常灵活。
但是,这是一个预先排序的列表,因此不适合您的示例。此外,我一直想用一种标准的树结构和 HTML 的 <ul> 和 <li> 元素来解决这个问题。
我提出的基本概念如下:
-
TreeNode - 将每个元素抽象为一个简单的TreeNode 类型,该类型可以提供它的值(例如Name)以及它是否有子元素。
-
TreeNodesIterator - 一个RecursiveIterator,能够遍历这些TreeNodes 的集合(数组)。这相当简单,因为 TreeNode 类型已经知道它是否有孩子以及哪些孩子。
-
RecursiveListIterator - 一个RecursiveIteratorIterator,它具有递归迭代任何类型的RecursiveIterator 时所需的所有事件:
-
beginIteration / endIteration - 主列表的开始和结束。
-
beginElement / endElement - 每个元素的开始和结束。
-
beginChildren / endChildren - 每个子列表的开始和结束。
这个RecursiveListIterator 仅以函数调用的形式提供这些事件。子列表,因为它是典型的 <ul><li> 列表,在其父 <li> 元素内打开和关闭。因此,endElement 事件在相应的endChildren 事件之后触发。这可以更改或可配置以扩大此类的使用。这些事件作为函数调用分发给装饰器对象,然后将它们分开。
-
ListDecorator - 一个“装饰器”类,它只是RecursiveListIterator 事件的接收者。
我从主输出逻辑开始。采用现在分层的$tree 数组,最终代码如下所示:
$root = new TreeNode($tree);
$it = new TreeNodesIterator(array($root));
$rit = new RecursiveListIterator($it);
$decor = new ListDecorator($rit);
$rit->addDecorator($decor);
foreach($rit as $item)
{
$inset = $decor->inset(1);
printf("%s%s\n", $inset, $item->getName());
}
首先让我们看一下ListDecorator,它简单地包装了<ul> 和<li> 元素并决定如何输出列表结构:
class ListDecorator
{
private $iterator;
public function __construct(RecursiveListIterator $iterator)
{
$this->iterator = $iterator;
}
public function inset($add = 0)
{
return str_repeat(' ', $this->iterator->getDepth()*2+$add);
}
构造函数采用它正在处理的列表迭代器。 inset 只是一个帮助函数,用于很好地缩进输出。其余的只是每个事件的输出函数:
public function beginElement()
{
printf("%s<li>\n", $this->inset());
}
public function endElement()
{
printf("%s</li>\n", $this->inset());
}
public function beginChildren()
{
printf("%s<ul>\n", $this->inset(-1));
}
public function endChildren()
{
printf("%s</ul>\n", $this->inset(-1));
}
public function beginIteration()
{
printf("%s<ul>\n", $this->inset());
}
public function endIteration()
{
printf("%s</ul>\n", $this->inset());
}
}
考虑到这些输出功能,这又是主要的输出总结/循环,我一步一步来:
$root = new TreeNode($tree);
创建将用于开始迭代的根TreeNode:
$it = new TreeNodesIterator(array($root));
这个TreeNodesIterator 是一个RecursiveIterator,它可以在单个$root 节点上进行递归迭代。它作为数组传递,因为该类需要迭代一些东西并允许与一组子元素重用,这些子元素也是TreeNode 元素的数组。
$rit = new RecursiveListIterator($it);
这个RecursiveListIterator 是一个提供上述事件的RecursiveIteratorIterator。要使用它,只需要提供一个ListDecorator(上面的类)并分配addDecorator:
$decor = new ListDecorator($rit);
$rit->addDecorator($decor);
然后一切都设置为仅foreach 并输出每个节点:
foreach($rit as $item)
{
$inset = $decor->inset(1);
printf("%s%s\n", $inset, $item->getName());
}
如本例所示,整个输出逻辑被封装在ListDecorator 类和单个foreach 中。整个递归遍历已经完全封装在 SPL 递归迭代器中,它提供了一个堆栈过程,这意味着内部没有进行递归函数调用。
基于事件的ListDecorator 允许您专门修改输出并为同一数据结构提供多种类型的列表。甚至可以更改输入,因为数组数据已封装到 TreeNode。
完整代码示例:
<?php
namespace My;
$tree = array('H' => 'G', 'F' => 'G', 'G' => 'D', 'E' => 'D', 'A' => 'E', 'B' => 'C', 'C' => 'E', 'D' => null);
// add children to parents
$flat = array(); # temporary array
foreach ($tree as $name => $parent)
{
$flat[$name]['name'] = $name; # self
if (NULL === $parent)
{
# no parent, is root element, assign it to $tree
$tree = &$flat[$name];
}
else
{
# has parent, add self as child
$flat[$parent]['children'][] = &$flat[$name];
}
}
unset($flat);
class TreeNode
{
protected $data;
public function __construct(array $element)
{
if (!isset($element['name']))
throw new InvalidArgumentException('Element has no name.');
if (isset($element['children']) && !is_array($element['children']))
throw new InvalidArgumentException('Element has invalid children.');
$this->data = $element;
}
public function getName()
{
return $this->data['name'];
}
public function hasChildren()
{
return isset($this->data['children']) && count($this->data['children']);
}
/**
* @return array of child TreeNode elements
*/
public function getChildren()
{
$children = $this->hasChildren() ? $this->data['children'] : array();
$class = get_called_class();
foreach($children as &$element)
{
$element = new $class($element);
}
unset($element);
return $children;
}
}
class TreeNodesIterator implements \RecursiveIterator
{
private $nodes;
public function __construct(array $nodes)
{
$this->nodes = new \ArrayIterator($nodes);
}
public function getInnerIterator()
{
return $this->nodes;
}
public function getChildren()
{
return new TreeNodesIterator($this->nodes->current()->getChildren());
}
public function hasChildren()
{
return $this->nodes->current()->hasChildren();
}
public function rewind()
{
$this->nodes->rewind();
}
public function valid()
{
return $this->nodes->valid();
}
public function current()
{
return $this->nodes->current();
}
public function key()
{
return $this->nodes->key();
}
public function next()
{
return $this->nodes->next();
}
}
class RecursiveListIterator extends \RecursiveIteratorIterator
{
private $elements;
/**
* @var ListDecorator
*/
private $decorator;
public function addDecorator(ListDecorator $decorator)
{
$this->decorator = $decorator;
}
public function __construct($iterator, $mode = \RecursiveIteratorIterator::SELF_FIRST, $flags = 0)
{
parent::__construct($iterator, $mode, $flags);
}
private function event($name)
{
// event debug code: printf("--- %'.-20s --- (Depth: %d, Element: %d)\n", $name, $this->getDepth(), @$this->elements[$this->getDepth()]);
$callback = array($this->decorator, $name);
is_callable($callback) && call_user_func($callback);
}
public function beginElement()
{
$this->event('beginElement');
}
public function beginChildren()
{
$this->event('beginChildren');
}
public function endChildren()
{
$this->testEndElement();
$this->event('endChildren');
}
private function testEndElement($depthOffset = 0)
{
$depth = $this->getDepth() + $depthOffset;
isset($this->elements[$depth]) || $this->elements[$depth] = 0;
$this->elements[$depth] && $this->event('endElement');
}
public function nextElement()
{
$this->testEndElement();
$this->event('{nextElement}');
$this->event('beginElement');
$this->elements[$this->getDepth()] = 1;
}
public function beginIteration()
{
$this->event('beginIteration');
}
public function endIteration()
{
$this->testEndElement();
$this->event('endIteration');
}
}
class ListDecorator
{
private $iterator;
public function __construct(RecursiveListIterator $iterator)
{
$this->iterator = $iterator;
}
public function inset($add = 0)
{
return str_repeat(' ', $this->iterator->getDepth()*2+$add);
}
public function beginElement()
{
printf("%s<li>\n", $this->inset(1));
}
public function endElement()
{
printf("%s</li>\n", $this->inset(1));
}
public function beginChildren()
{
printf("%s<ul>\n", $this->inset());
}
public function endChildren()
{
printf("%s</ul>\n", $this->inset());
}
public function beginIteration()
{
printf("%s<ul>\n", $this->inset());
}
public function endIteration()
{
printf("%s</ul>\n", $this->inset());
}
}
$root = new TreeNode($tree);
$it = new TreeNodesIterator(array($root));
$rit = new RecursiveListIterator($it);
$decor = new ListDecorator($rit);
$rit->addDecorator($decor);
foreach($rit as $item)
{
$inset = $decor->inset(2);
printf("%s%s\n", $inset, $item->getName());
}
输出:
<ul>
<li>
D
<ul>
<li>
G
<ul>
<li>
H
</li>
<li>
F
</li>
</ul>
</li>
<li>
E
<ul>
</li>
<li>
A
</li>
<li>
C
<ul>
<li>
B
</li>
</ul>
</li>
</ul>
</li>
</ul>
</li>
</ul>
Demo (PHP 5.2 variant)
一个可能的变体是迭代任何RecursiveIterator 并提供对所有可能发生的事件的迭代的迭代器。然后 foreach 循环中的 switch / case 可以处理这些事件。
相关: