PHP 递归迭代器迭代器 : Determining first and last item at each branch level

Question

我扩展了 Zend_View_Helper_Navigation_Menu，它使用 RecursiveIteratorIterator 来迭代菜单树。我希望能够确定的是我是在树中分支级别的第一项还是最后一项。

这是我正在寻找的示例:

• 导航 1(第一个)
  • Nav 1.1(第一个和最后一个)
    • 导航 1.1.1(第一)
    • 导航 1.1.2
    • 导航 1.1.3(最后)
• 导航 2
  • 导航 2.1(第一)
  • 导航 2.2(最后)
• 导航 3(最后)
  • 导航 3.1(第一)
  • 导航 3.2(最后)

---

附加信息

• PHP 版本 5.2.13

---

解决方案

在 foreach ($iterator as $page) 循环中，可以使用两个变量来跟踪深度，$depth 和 $prevDepth。然后，一个简单的比较条件就可以确定分支级别中的第一项:if ($depth > $prevDepth)。

使用 Zend_Navigation_Container 对象创建一个 RecursiveCachingIterator，然后使用它来创建 RecursiveIteratorIterator 添加 hasNext() 方法。

$rci = new RecursiveCachingIterator($container, CachingIterator::FULL_CACHE);
$iterator = new RecursiveIteratorIterator($rci,
                    RecursiveIteratorIterator::SELF_FIRST);
/* snip */
$prevDepth = -1;
foreach ($iterator as $page) {
    $depth = $iterator->getDepth();
    /* snip */
    if ($depth > $prevDepth) {
        // first branch item
    }
    /* snip */
    if (!$iterator->hasNext()) {
        // last branch item
    }
    /* snip */
    $prevDepth = $depth;
}

Answer 1

使用递归缓存迭代器:

$rdi = new RecursiveDirectoryIterator('.');
$rci = new RecursiveCachingIterator($rdi, CachingIterator::FULL_CACHE); 
$rii = new RecursiveIteratorIterator($rci, RecursiveIteratorIterator::SELF_FIRST);

foreach ($rii as $file) {
    if ($file->isDir()) {
        echo $file->getFilename() . PHP_EOL;
    }
    elseif (!$rii->hasNext()) {
        echo $file->getFilename() . PHP_EOL;
    }
    elseif (count($rii->getCache()) == 1) {
        echo $file->getFilename() . PHP_EOL;
    }
}

另一个数组解决方案:

function buildTree(RecursiveDirectoryIterator $iterator) {
    $tree = array();
    foreach ($iterator as $fileinfo) {
        if ($fileinfo->isDir()) {
            $tree[$fileinfo->getFilename()] = buildTree($iterator->getChildren());
        } else {
            $tree[$fileinfo->getFilename()] = $fileinfo->getFilename();
        }
    }
    return $tree;
}

function filterTree(array $tree) {
    foreach ($tree as $key => $value) {
        if (is_array($value)) {
            $tree[$key] = filterTree($value);
        } elseif (reset($tree) !== $value && end($tree) !== $value) {
            unset($tree[$key]);
        }
    }
    return $tree;
}

print_r(filterTree(buildTree(new RecursiveDirectoryIterator('.'))));