Binary Tree Traverasl with OO and Stack

1、一切都是對象，對象就是關(guān)系和操作；每個節(jié)點，都有兩種操作，print和visit。 2、放在stack里面可以使操作，可以是函數(shù)，遞歸也是面向?qū)ο蟮捏w現(xiàn)；

Python代碼實現(xiàn)如下：

class guide(object):def __init__(self, opt,node):self.opt = optself.node = nodeclass Solution2(object):def preorderTraversal(self, root):""":type root: TreeNode:rtype: List[int]"""if root is None:return []ans = []path = [guide(0,root),]while path:current = path.pop()if not current.node:continueif current.opt == 1:ans += [current.node.item]else:path += [guide(0,current.node.right)]path += [guide(0,current.node.left)]path += [guide(1,current.node)]return ansdef inorderTraversal(self, root):""":type root: TreeNode:rtype: List[int]"""if root is None:return []ans = []path = [guide(0,root),]while path:current = path.pop()if not current.node:continueif current.opt == 1:ans += [current.node.item]else:path += [guide(0,current.node.right)]path += [guide(1,current.node)]path += [guide(0,current.node.left)]return ansdef postorderTraversal(self, root):""":type root: TreeNode:rtype: List[int]"""if root is None:return []ans = []path = [guide(0,root),]while path:current = path.pop()if not current.node:continueif current.opt == 1:ans += [current.node.item]else:path += [guide(1,current.node)]path += [guide(0,current.node.right)]path += [guide(0,current.node.left)]return ans

Test case：

if __name__ =="__main__":tree =tree.Node()tree._add(0)tree._add(1)tree._add(2)tree._add(3)tree._add(4)tree._add(5)tree._add(6)tree._add(7)tree._add(8)tree._add(9)tree._add(10)ans = Solution3().preorderTraversal(tree)print ansans = Solution3().inorderTraversal(tree)print ansans = Solution3().postorderTraversal(tree)print ans

Output:

[0, 1, 3, 5, 7, 9, 10, 8, 6, 4, 2] [9, 7, 10, 5, 8, 3, 6, 1, 4, 0, 2] [9, 10, 7, 8, 5, 6, 3, 4, 1, 2, 0]

總結(jié)

以上是生活随笔為你收集整理的Binary Tree Traverasl with OO and Stack的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。