LeetCode 589. N 叉树的前序遍历

见解分享

2023-10-05 01:26:56

引言

在计算机科学中，N 叉树是一种数据结构，其中每个节点可以拥有任意数量的子节点。前序遍历是一种遍历 N 叉树的算法，其中根节点首先被访问，然后是它的子节点，再是它们的子节点，依此类推。

题目

LeetCode 589. N 叉树的前序遍历要求我们实现一个函数来返回 N 叉树的前序遍历结果。给定一个 N 叉树的根节点，我们必须返回该树的前序遍历顺序中的节点值列表。

解决方案

我们可以使用两种主要方法来解决此问题：

1. 递归方法

递归方法遵循分而治之的原则。算法如下：

/**
 * Definition for Node.
 * class Node {
 *     public $val;
 *     public $children;
 *     function __construct($val) { $this->val = $val; $this->children = []; }
 * }
 */

function preorder($root) {
    if (!$root) {
        return [];
    }

    $result = [$root->val];
    foreach ($root->children as $child) {
        $result = array_merge($result, preorder($child));
    }

    return $result;
}

2. 迭代方法

迭代方法使用栈来跟踪要访问的节点。算法如下：

function preorder($root) {
    if (!$root) {
        return [];
    }

    $stack = [$root];
    $result = [];

    while (!empty($stack)) {
        $node = array_pop($stack);
        $result[] = $node->val;

        foreach (array_reverse($node->children) as $child) {
            array_push($stack, $child);
        }
    }

    return $result;
}