trampoline

Python 的递归是有最大限制的,超过这个限制便会抛出

RuntimeError: maximum recursion depth exceeded

sys.getrecursionlimit()可以查看递归的最大层数,默认为 1000

sys.setrecursionlimit() 可以改变这个最大层数

cPython 不支持尾递归优化,但有些 hack 的做法可以绕过,比如 trampoline 这种神奇的技术。当调用递归函数时,先插入一个 trampoline 函数。通过这个函数来调用真正的递归函数,并且对其进行修改,不让它再次进行递归的函数调用,而是直接返回下次递归调用的参数,由 trampoline 来进行下一次“递归“调用。这样一层一层的递归调用就会变成一次一次的迭代调用。

个人认为 如果要使用这种 hack 的做法,不如去使用一种可以对尾递归进行优化的语言

参考自http://code.activestate.com/recipes/474088/,有少许改动

#!/usr/bin/env python2.7
# This program shows off a python decorator(
# which implements tail call optimization. It
# does this by throwing an exception if it is
# it's own grandparent, and catching such
# exceptions to recall the stack.

import sys
from functools import wraps


class TailRecurseException(BaseException):

    def __init__(self, args, kwargs):
        self.args = args
        self.kwargs = kwargs


def tail_call_optimized(func):
    """
    This function decorates a function with tail call
    optimization. It does this by throwing an exception
    if it is it's own grandparent, and catching such
    exceptions to fake the tail call optimization.

    This function fails if the decorated
    function recurses in a non-tail context.
    """
    @wraps(func)
    def wrapper(*args, **kwargs):
        f = sys._getframe()
        if f.f_back and f.f_back.f_back \
                and f.f_back.f_back.f_code == f.f_code:
            raise TailRecurseException(args, kwargs)
        else:
            while True:
                try:
                    return func(*args, **kwargs)
                except TailRecurseException as e:
                    args = e.args
                    kwargs = e.kwargs
    return wrapper


@tail_call_optimized
def factorial(n, acc=1):
    "calculate a factorial"
    if n == 0:
        return acc
    return factorial(n - 1, n * acc)


print factorial(10000)
# prints a big, big number,
# but doesn't hit the recursion limit.


@tail_call_optimized
def fib(i, current=0, next=1):
    if i == 0:
        return current
    else:
        return fib(i - 1, next, current + next)


print fib(10000)
# also prints a big number,
# but doesn't hit the recursion limit.

这段代码最精彩地方的就是一发生递归就会抛出一个携带函数参数的自定义异常,然后通过循环的方式来再次调用(传入捕获到的参数)。这样我们虽然写代码时写的是递归,但是实际上发生的迭代。

上面的代码使用了 sys._getframe() 获取栈帧,以判断是否发生递归。这个平时写代码时一般很少接触到。

Python 文档中是这样表述的

Return a frame object from the call stack. If optional integer depth is given, return the frame object that many calls below the top of the stack. If that is deeper than the call stack, ValueError is raised. The default for depth is zero, returning the frame at the top of the call stack.

CPython implementation detail: This function should be used for internal and specialized purposes only. It is not guaranteed to exist in all implementations of Python.

# 获取当前栈帧对象
f = sys._getframe()
# 获取前一个栈帧对象
f.f_back
# 获取栈帧的代码对象
f.f_code
# 获取栈帧的代码对象名称
f.f_code.co_name

另外当没有前一个栈帧时,f.f_back 会返回 None

回到之前的代码中,大致分析一下流程
执行 factorial(1000) ,由于我们使用了tail_call_optimized 装饰器,实际上我们调用了 wrapper
这时前一个栈帧的代码对象名称为 <module> (这里表达的有点不恰当,大概是这个意思),前前栈帧的代码对象名称为 execfile
进入 while True 死循环,执行 func(*args, **kwargs) 创建了新的栈帧。这时发生递归调用又进入新的 wrapper 栈帧中,但是栈帧对应的代码对象(f_code)不变,所以这里我们抛出了自定义异常 TailRecurseException ,并将此次递归的参数写入
捕获到这个异常后,并用从异常对象中读出的参数再次执行 func(*args, **kwargs)

注意,我们并没有阻止栈帧的线性增长。

另外还有一种利用 yield 的实现

http://www.jianshu.com/p/d36746ad845d

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 151,829评论 1 331
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 64,603评论 1 273
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 101,846评论 0 226
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 42,600评论 0 191
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 50,780评论 3 272
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 39,695评论 1 192
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,136评论 2 293
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 29,862评论 0 182
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 33,453评论 0 229
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 29,942评论 2 233
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,347评论 1 242
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 27,790评论 2 236
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,293评论 3 221
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 25,839评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,448评论 0 181
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 34,564评论 2 249
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 34,623评论 2 249

推荐阅读更多精彩内容