给定一个 N 叉树,返回其节点值的前序遍历。
例如,给定一个 3叉树 :
返回其前序遍历: [1,3,5,6,2,4]。
说明: 递归法很简单,你可以使用迭代法完成此题吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal
思路:
迭代版:前序遍历的顺序是 [根,儿子].使用数组来模拟栈,来得到前序遍历。首先将根节点入栈,根节点是前序遍历中的第一个节点,然后从取出栈顶元素u,例如 u 的子节点从左到右为v1,v2,v3。因为栈的FILO的特点,需要逆序子节点以v3,v2,v1的顺序入栈,保证按照正确的出栈顺序:v1,v2,v3。
递归版:看代码吧。
复杂度分析:
时间复杂度O(n)
空间复杂度O(n)
//迭代版var preorder = function (root) {let ans = [];if (root == null) {return ans;}let stack = [];stack.push(root);while (stack.length) {let node = stack.pop();ans.push(node.val);// Collections.reverse(node.children);node.children.reverse();for (let item of node.children) {stack.push(item);}}return ans;};//递归版var preorder = function (root) {let ans = [];const dfs = (root) => {if (root === null) return;ans.push(root.val);for (let item of root.children) {dfs(item);}};dfs(root);return ans;};
