Python 生成器函数迭代

想要通过迭代实现二叉树的遍历，刚好刚刚看到Python中的yield。yield 可以将函数变成一个生成器（generator），并且返回一个iterabel对象。能过避免通过列表保存结果导致的内存占用。嘿嘿嘿，这么好的东西还不赶快尝试一下。

 

def preTrav_yield(root):
    if root==None:return
    yield  root.val
    preTrav_yield(root.left)
    preTrav_yield(root.right)

用最常见的前序遍历的函数迭代格式，只是把print 换成了yield。美滋滋的开始测试。

if __name__=="__main__":
    tree=TreeNode(1,TreeNode(3,TreeNode(5,TreeNode(8),\
TreeNode(6)),TreeNode(4)),TreeNode(15,TreeNode(44),TreeNode(86)))
    print("test preTraverse:")
    for i in preTrav_yield(tree):
        print(i)

运行结果却有点出乎预料......

只输出了根节点......

这时我注意到了yield的几个属性：

1. 当调用生成器函数时，函数将返回一个对象，但是不会立刻向下执行
2. 像__iter__()和__next__()方法等是自动实现的，所以我们可以通过next()方法对对象进行迭代
3. 一旦函数被yield，函数会暂停，控制权返回调用者
4. 局部变量和它们的状态会被保存，直到下一次调用

这就意味着：当我们以函数名（形参）的形式调用生成器函数时，函数只会执行到yield语句，其后的语句不会立即执行，只有当生成器接收到__next()__方法时，函数才会继续运行（for循环会隐式的调用__next()__函数）。因此在我写的函数中preTrav_yield(root.left)  这一句不会像其他函数一样自动进行迭代，而是等待__next()__函数的调用以继续运行。于是对代码进行修改。修改后得代码如下：

def preTrav_yield(root):
    if root==None:return
    yield root.val
    for i in preTrav_yield(root.left):
        yield i
    for i in preTrav_yield(root.right):
        yield i

测试结果：

这样就可以输出正确的结果了。

 

小白学习Python的笔记，如果大神发现有不对的地方，还请多多指教。