给定一个 N 叉树,返回其节点值的 后序遍历
N 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。

示例

示例 1:
image.png
输入:root = [1,null,3,2,4,null,5,6]
输出:[5,6,3,2,4,1]

示例 2:

输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[2,6,14,11,7,3,12,8,4,13,9,10,5,1]

题解

建立一个栈,遍历 N 叉树,取出该节点的第一个子节点推入栈
stack = [①] res = []
stack = [①③] res = []
stack = [①③⑤] res =[]

等到节点⑤没有子节点,推出⑤,记录数据5,指针再指向推出的③
stack = [①] res = [5]

将③推入栈,继续遍历第一个字节点⑥推入栈
stack = [①③⑥] res =[5]

节点⑥没有子节点,记录数据6,指针再指向推出的③
stack = [①] res =[5,6]

节点③已经没有子节点,记录数据3,,指针再指向推出的①
stack = [] res =[5,6,3]

遍历节点①,推入①和它的第一个子节点②
stack = [①②] res =[5,6,3]

节点耳没有子节点,记录数据2,重复以上
stack = [] res =[5,6,3,2]

  1. /**
  2. * // Definition for a Node.
  3. * function Node(val,children) {
  4. * this.val = val;
  5. * this.children = children;
  6. * };
  7. */
  8. /**
  9. * @param {Node|null} root
  10. * @return {number[]}
  11. */
  12. var postorder = function(root) {
  13. if (!root) {
  14. return []
  15. }
  16. const stack = []
  17. let cur = root
  18. const res = []
  19. while(stack.length || cur) {
  20. if (cur) {
  21. stack.push(cur)
  22. cur = (cur.children && cur.children.length > 0) ? cur.children.shift() : null
  23. } else {
  24. const node = stack.pop()
  25. res.push(node.val)
  26. cur = stack.pop()
  27. }
  28. }
  29. return res
  30. };

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。