589. N-ary Tree Preorder Traversal & 590. N-ary Tree Postorder Traversal

65 篇文章 0 订阅
26 篇文章 0 订阅

Given an n-ary tree, return the preorder traversal of its nodes' values.

For example, given a 3-ary tree:

 

Return its preorder traversal as: [1,3,5,6,2,4].

Note: Recursive solution is trivial, could you do it iteratively?

 

 

 

 

Given an n-ary tree, return the postorder traversal of its nodes' values.

For example, given a 3-ary tree:

 

Return its postorder traversal as: [5,6,3,2,4,1].

Note: Recursive solution is trivial, could you do it iteratively?

 

 

 

循环写法不太好用之前二叉树的方法写:https://blog.csdn.net/zjucor/article/details/53447464

因为一直往左边遍历完自后,我是知道要往右边的,而且直接root.right就访问到了,但是在这里是不行的,遍历完第1个child Node,是无法直接得到下个要遍历的节点。

所以要一次性就把children加到stack的解法


class Solution(object):
    def preorder(self, root):
        """
        :type root: Node
        :rtype: List[int]
        """
        if not root: return []
        st=[root]
        res=[]
        while st:
            r=st.pop()
            res.append(r.val)
            st += r.children[::-1]
        return res
        

class Solution(object):
    def postorder(self, root):
        """
        :type root: Node
        :rtype: List[int]
        """
        if not root: return []
        res, st = [], [root]
        while st:
            r=st.pop()
            res.append(r.val)
            st+=r.children
        return res[::-1]

这种写法也是super simple啊

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值