php - 使用 RecursiveArrayIterator 获取所有父节点

Question

本质上，我想使用

$foo = new RecursiveIteratorIterator(new RecursiveArrayIterator($haystack));

方法论，但不是返回一个平面数组，foreach()而是保留结构，但只返回一个曾孙节点及其父节点。这在PHP中可能吗？

我的任务是优化我公司的一些（可怕的）代码库。我找到了一个通过数组递归搜索键的函数。我不能用简单的array_search()or替换它，array_key_exists()因为自定义函数还返回匹配（找到）键的父节点，而不仅仅是 a trueor false。

我如何使用或失败RecursiveArrayIterator，RecursiveIteratorIterator其他内置函数（即尽可能少的自定义代码）从搜索数组返回与其父树匹配的节点？我想获得最快的函数，因为目前这个函数需要 8 秒执行 14000 次，因此我需要使用内置函数。

我现有的尝试（下）非常慢。

function search_in_array($needle, $haystack) {
    $path = array();

    $it = new RecursiveArrayIterator($haystack);
    iterator_apply($it, 'traverse', array($it, $needle, &$path));

    return $path;
}

function traverse($it, $needle, &$path) {
    while($it->valid()) {
        $key = $it->key();
        $value = $it->current();

        if(strcasecmp($value['id'], $needle) === 0) {
            $path[] = $key;
            return;
        } else if($it->hasChildren()) {
            $sub = null;
            traverse($it->getChildren(), $needle, &$sub);
            if($sub) {
                $path[$key] = $sub;
            }
        }

        $it->next();
    }
}

的示例输出$needle = TVALL如下所示：

Array (
    [HOMECINEMA] => Array (
        [children] => Array (
            [HC-VISION] => Array (
                [children] => Array (
                    [0] => TVALL
                )
            )
        )
    )
)

搜索数组看起来像这样（对不起，太庞大了）。有两个以上的顶级节点，但为简洁起见，我将其截断：

score 7 · Accepted Answer

下面的示例不一定具有更高的性能（在时间或内存要求方面），但避免了手动递归遍历结构并显示了一种更简单的（恕我直言）方式来构建所需的输出数组。

function search_in_array($needle, $haystack) {
    $path = array();

    $it = new RecursiveIteratorIterator(
        new ParentIterator(new RecursiveArrayIterator($haystack)),
        RecursiveIteratorIterator::SELF_FIRST
    );

    foreach ($it as $key => $value) {
        if (array_key_exists('id', $value) && strcasecmp($value['id'], $needle) === 0) {
            $path = array($needle);
            for ($i = $it->getDepth() - 1; $i >= 0; $i--) {
                $path = array($it->getSubIterator($i)->key() => $path);
            }
            break;
        }
    }

    return $path;
}

参考

RecursiveIteratorIterator::SELF_FIRST迭代模式 - 需要查看非“叶”项
ParentIterator - 过滤掉“叶子”项目的简单方法
RecursiveIteratorIterator::getDepth()
RecursiveIteratorIterator::getSubIterator()

奖金

如果您的数组也更深，您还可以使用RecursiveIteratorIterator::setMaxDepth()方法将递归限制在深度级别。n

php - 使用 RecursiveArrayIterator 获取所有父节点

1 回答 1

Related

Reference