递归和引用传递

Question

我有一个具有以下结构的类别树：

[6] => Array
    (
        [id] => 6
        [name] => computers
        [productCount] => 0
        [children] => Array
            (
                [91] => Array
                    (
                        [id] => 91
                        [name] => notebook
                        [productCount] => 5
                        [children] => Array
                            (
                            )
                    )

                [86] => Array
                    (
                        [id] => 86
                        [name] => desktop
                        [productCount] => 0
                        [children] => Array
                            (
                            )
                    )
            )
    )

除了子类别之外，每个类别都可能包含产品（就像一个文件夹可能包含子文件夹而仅包含文件）。

我正在尝试编写一个递归函数，我想将此数组作为参考，并剥离具有 [productCount] = 0 的叶类别和包含此类空节点的所有父类别。换句话说，在处理之后，我只想拥有那些在任何子级别上包含产品的类别。

我已经写了一些代码，现在正在调试它，它不会剥离空节点。可能是我没有正确使用参考。如果可能，请帮我解决它。

    function pruneTree( & $node) {
    if ( ! $node['children'] && ! $node['productCount']) {
        unset($node);
    }
    if ( ! empty($node['children'])) {
        foreach ($node['children'] as $key => $child) {
            pruneTree($node['children'][$key]);
        }
    }
    return;
}

Answer 1

我会那样做。注意 foreach 中的“&”。

function pruneTree(&$node)
{
    foreach ($node as $index => &$value) {
        if (empty($value)) {
            unset($node[$index]);
        } elseif (is_array($value)) {
            pruneTree($value);
        }
    }
}

Answer 2

我不知道是不是这样，但是当我需要在数组中递归更改值时，我也需要将 & 传递给 foreach 值。

private function convertXMLPart(&$array) {
        foreach ($array as $rowKey => &$row) {
            if (gettype($row) != 'string') {
                $row = (array)$row;
                if (!empty($row['@attributes'])) {
                    foreach ($row['@attributes'] as $key => $value) {
                        $row[$key] = $value;
                    }
                    unset($row['@attributes']);
                    $array[$rowKey] = $row;
                }
                $this->convertXMLPart($row);
            }
        }
    }

Answer 3

您还可以更改函数中的参数以获取节点数组而不是单个节点。这会稍微改变递归，并避免传递密钥：

function pruneTree(&$nodes) {
    foreach ($nodes as $key => $node) {
        if (!$node['children'] && !$node['productCount']) {
            unset($nodes[$key]);
        } elseif (!empty($node['children'])) {
            pruneTree($nodes[$key]['children']);
            // This line checks if all the children have been pruned away:
            if (empty($nodes[$key]['children'])) {
                unset($nodes[$key]);
            }
        }
    }
}

此外，添加了一项检查，以确保如果所有子节点都被修剪，则父（现在，叶）节点也会被修剪。

希望这会有所帮助！

---

测试数据：

$data = array(
    6 => array(
        'id' => 6,
        'name' => 'computers',
        'productCount' => 0,
        'children' => array(
            91 => array(
                'id' => 91,
                'name' => 'notebook',
                'productCount' => 5,
                'children' => array()
            ),
            86 => array(
                'id' => 86,
                'name' => 'desktop',
                'productCount' => 0,
                'children' => array()
            )
        )
    )
);

召唤：

pruneTree($data);
echo '<pre>';
print_r($data);
echo '</pre>';

Answer 4

unset 只删除引用而不删除被引用的变量：

如果通过引用传递的变量是函数内部的unset()，则只有局部变量被销毁。调用环境中的变量将保留与调用 unset() 之前相同的值。

所以你需要传递父数组和删除该变量的键：

function pruneTree(&$parent, $key) {
    $node = &$parent[$key];
    if (!$node['children'] && !$node['productCount']) {
        unset($parent[$key]);
    }
    if (!empty($node['children'])) {
        foreach ($node['children'] as $key => &$child) {
            pruneTree($node['children'], $key);
        }
    }
}