题解 | #实现二叉树先序,中序和后序遍历#

实现二叉树先序,中序和后序遍历

http://www.nowcoder.com/practice/a9fec6c46a684ad5a3abd4e365a9d362

代码如下,第一次发代码上来,我还是个小白
大神们告诉我我这种分开定义函数的方法是不是很low。。。。。

看了大神们的解法,发现我没有理解递归的妙处,stuck和res两个变量相当于画蛇添足。用循环迭代才需要这两个变量,重写一遍对递归有了更深刻的理解,新代码放在原来这个后面。

两个对比来看,方便理解
import copy
class Solution:
def threeOrders(self , root ):
res=[]
stuck=[]
res2=[]
stuck2=[]
res3=[]
stuck3=[]
res1= self.find(root,res,stuck)
res2= self.find2(root,res2,stuck2)
res3= self.find3(root,res3,stuck3)
return (res1,res2,res3)
# write code here
def find(self,root,res,stuck):
if root is None:
return
if root.right is not None:
stuck.append(root.right)
res.append(root.val)
self.find(root.left,res,stuck)
if len(stuck)!=0:
root=stuck.pop()
self.find(root,res,stuck)
return res
def find2(self,root,res2,stuck2):
if root is None:
return
stuck2.append(root)
self.find2(root.left,res2,stuck2)
if len(stuck2)!=0:
root=stuck2.pop()
res2.append(root.val)
self.find2(root.right,res2,stuck2)
return res2
def find3(self,root,res3,stuck3):
if root is None:
return
stuck3.append(root)
self.find3(root.left,res3,stuck3)
if len(stuck3)!=0:
temp=stuck3.pop()
root=copy.copy(temp)
self.find3(root.right,res3,stuck3)
res3.append(temp.val)
return res3
新代码~
import copy
class Solution:
def threeOrders(self , root ):
res1,res2,res3=[],[],[]
res=[res1,res2,res3]
res1=self.find(root,res)
return (res1)
# write code here
def find(self,root,res):
if root is None:
return
res[0].append(root.val)
self.find(root.left,res)
res[1].append(root.val)
self.find(root.right,res)
res[2].append(root.val)
return res

全部评论

相关推荐

10-30 22:18
已编辑
毛坦厂中学 C++
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务