尾呼叫
在電腦學裡,尾呼叫是指一個函數裡的最後一個動作是返回一個函式的呼叫結果的情形,即最後一步新呼叫的返回值直接作為當前函式的返回結果。[1]此時,該尾部呼叫位置被稱為尾位置。尾呼叫中有一種重要而特殊的情形叫做尾遞迴。經過適當處理,尾遞迴形式的函式的執行效率可以被極大地最佳化。[1]尾呼叫原則上都可以通過簡化函式呼叫棧的結構而獲得效能最佳化(稱為「尾呼叫消除」),但是最佳化尾呼叫是否方便可行取決於執行環境對此類最佳化的支援程度如何。
概述
在電腦科學裡,尾呼叫是指一個函數裡的最後一個動作是一個函式呼叫的情形:即這個呼叫的返回值直接被當前函式返回的情形。這種情形下稱該呼叫位置為尾位置。若這個函式在尾位置呼叫本身(或是一個尾呼叫本身的其他函式等等),則稱這種情況為尾遞迴,是遞迴的一種特殊情形。尾呼叫不一定是遞迴呼叫,但是尾遞迴特別有用,也比較容易實現。
在程式執行時,電腦會為應用程式分配一定的主記憶體空間;應用程式則會自行分配所獲得的主記憶體空間,其中一部分被用於記錄程式中正在呼叫的各個函式的執行情況,這就是函式的呼叫棧。常規的函式呼叫總是會在呼叫棧最上層添加一個新的堆疊框(stack frame,也翻譯為「堆疊框」或簡稱為「訊框」),這個過程被稱作「入棧」或「壓棧」(意即把新的訊框壓在棧頂)。當函式的呼叫層數非常多時,呼叫棧會消耗不少主記憶體,甚至會撐爆主記憶體空間(棧溢位)[1],造成程式嚴重卡頓或意外崩潰。尾呼叫的呼叫棧則特別易於最佳化,從而可減少主記憶體空間的使用,也能提高執行速度。[1]其中,對尾遞迴情形的最佳化效果最為明顯,尤其是遞迴演算法非常複雜的情形。[1]
一般來說,尾呼叫消除是可選的,可以用,也可以不用。然而,在函數式程式語言中,語言標準通常會要求編譯器或執行平台實現尾呼叫消除。這讓程式設計師可以用遞迴取代迴圈而不喪失效能。
定義與說明
定義
尾呼叫 (tail call) 指的是一個函式的最後一條語句也是一個返回呼用函式的語句。在函式體末尾被返回的可以是對另一個函式的呼叫,也可以是對自身呼叫(即自身遞迴呼叫)。[1]
特徵與簡單範例
尾呼叫可能位於一個函式語法上最後的位置:
function foo(data) {
a(data);
return b(data);
}
在這裡,a(data)
、b(data)
都是函式呼叫,但是 b(data)
是函式返回前的最後執行的東西,所以也是所謂的尾位置。然後,並非所有的尾呼叫都必須在一個函式語法上最後的位置。考慮:
function bar(data) {
if ( a(data) ) {
return b(data);
}
return c(data);
}
在這裡,b
、c
的呼叫都在尾位置。這是因為儘管 b(data)
不在 bar
語法上最後的位置,它是 if
敘述其中一個分支最後的位置。
現在考慮以下代碼:
function foo1(data) {
return a(data) + 1;
}
function foo2(data) {
var ret = a(data);
return ret;
}
function foo3(data) {
var ret = a(data);
return (ret === 0) ? 1 : ret;
}
在這裡,a(data)
處於 foo2
的尾位置,但不處於 foo1
或 foo3
的尾位置。這是因為程式必須返回這2個 a
函式的呼叫以檢查、更動 a
的返回值。
說明
傳統模式的編譯器對於尾呼叫的處理方式就像處理其他普通函式呼叫一樣,總會在呼叫時建立一個新的堆疊框(stack frame)並將其推入呼叫棧頂部,用於表示該次函式呼叫。[1]
當一個函式呼叫發生時,電腦必須 「記住」 呼叫函式的位置 —— 返回位置,才可以在呼叫結束時帶著返回值回到該位置,返回位置一般存在呼叫棧上。在尾呼叫這種特殊情形中,電腦理論上可以不需要記住尾呼叫的位置而從被呼叫的函式直接帶著返回值返回呼用函式的返回位置(相當於直接連續返回兩次)。尾呼叫消除即是在不改變當前呼叫棧(也不添加新的返回位置)的情況下跳到新函式的一種最佳化(完全不改變呼叫棧是不可能的,還是需要校正呼叫棧上形式參數與局部變數的資訊。[2])
由於當前函式訊框上包含局部變數等等大部分的東西都不需要了,當前的函式訊框經過適當的更動以後可以直接當作被尾呼叫的函式的訊框使用,然後程式即可以跳到被尾呼叫的函式。產生這種函式訊框更動代碼與 「jump」(而不是一般常規函式呼叫的代碼)的過程稱作尾呼叫消除(Tail Call Elimination)或尾呼叫最佳化(Tail Call Optimization,TCO)。尾呼叫最佳化讓位於尾位置的函式呼叫跟 goto
語句效能一樣高,也因此使得高效的結構編程成為現實。
然而,對於 C++ 等語言來說,在函式最後 return g(x); 並不一定是尾遞迴——在返回之前很可能涉及到對象的解構函式,使得 g(x) 不是最後執行的那個。這可以通過返回值最佳化來解決。
尾遞迴
若函式在尾位置呼叫自身(或是一個尾呼叫本身的其他函式等等),則稱這種情況為尾遞迴。尾遞迴也是遞迴的一種特殊情形。尾遞迴是一種特殊的尾呼叫,即在尾部直接呼叫自身的遞迴函式。對尾遞迴的最佳化也是關注尾呼叫的主要原因。尾呼叫不一定是遞迴呼叫,但是尾遞迴特別有用,也比較容易實現。
特點
尾遞迴在普通尾呼叫的基礎上,多出了2個特徵:
- 在尾部呼叫的是函式自身 (Self-called);
- 可通過最佳化,使得計算僅占用常數棧空間 (Stack Space)。
最佳化尾遞迴的分析與範例
對函式呼叫在尾位置的遞迴或互相遞迴的函式,由於函式自身呼叫次數很多,遞迴層級很深,尾遞迴最佳化則使原本 O(n) 的呼叫棧空間只需要 O(1)。因此一些程式語言的標準要求語言實現進行尾呼叫消除,例如 Scheme[3][4]與 ML 家族的語言。在 Scheme 中,語言標準還將尾位置形式化,指定了各種語法中允許尾調用的地方[5]。
以 Python 為例,主要區分普通遞迴和尾遞迴對棧空間的使用[6][需要較佳來源]:
def recsum(x):
if x == 1:
return x
else:
return x + recsum(x - 1)
呼叫recsum(5)
為例,SICP中描述了相應的棧空間變化[7]:
recsum(5)
5 + recsum(4)
5 + (4 + recsum(3))
5 + (4 + (3 + recsum(2)))
5 + (4 + (3 + (2 + recsum(1))))
5 + (4 + (3 + (2 + 1)))
5 + (4 + (3 + 3))
5 + (4 + 6)
5 + 10
15
可觀察,堆疊從左到右,增加到一個峰值後再計算從右到左縮小,這往往是我們不希望的,所以在C語言等語言中設計for, while, goto
等特殊結構語句,使用迭代、尾遞迴,對普通遞迴進行最佳化,減少可能對主記憶體的極端消耗。修改以上代碼,可以成為尾遞迴:
def tailrecsum(x, running_total=0):
if x == 0:
return running_total
else:
return tailrecsum(x - 1, running_total + x)
或者使用迭代:
for i in range(6):
sum += i
對比後者尾遞迴對主記憶體的消耗:
tailrecsum(5, 0)
tailrecsum(4, 5)
tailrecsum(3, 9)
tailrecsum(2, 12)
tailrecsum(1, 14)
tailrecsum(0, 15)
15
則是線性的。
最佳化尾呼叫的不同方式
要方便地實現尾呼叫最佳化,一般需藉助編譯器或執行環境提供的現成的尾遞迴最佳化特性,或是依賴所用程式語言能直接支援更底層的指令跳轉。
利用執行平台的支援直接實現
在 Perl 里,程式設計師可以直接用一種帶有函式名稱的 「goto」 敘述變體:goto &NAME;
直接使用尾呼叫[8]。
在程式語言實現中,消除尾遞迴里的尾呼叫比消除一般的尾呼叫容易很多。舉例來說,Java 虛擬機器(JVM)的實現會消除尾遞迴里的尾呼叫(因為重新使用了原來的呼叫棧),但是不會消除一般的尾呼叫(因為改變了的呼叫棧)。Scala 等同樣基於 JVM 平台的語言可以有效地實現單個函式的尾遞迴最佳化,但是對於多個函式的相互尾遞迴就無法最佳化了。
JavaScript則原本不支援尾呼叫最佳化,到其第6代語言核心標準「ECMAScript 6」開始規定程式引擎應在嚴格模式下使用尾呼叫最佳化。而且ECMAScript 6限定了尾位置不含閉包的尾呼叫才能進行最佳化。[1]
動手實現的方案
組譯重組
對於直接生成組譯的編譯器,尾部呼叫消除很簡單:只要校正棧上的形參之後把 「call」 的機器碼換成一個 「jump」 的就行了。從編譯器的觀點,以下代碼
function foo() return a()
先會被翻譯成(這是合法的 x86 組譯):
foo: call a ret
然後,尾部呼叫消除指的是將最後兩個指令以一個 「jump」 指令替換掉:
foo: jmp a
在 a
函式完成的時候,它會直接返回到 foo
的返回位址,省去了不必要的 ret
指令。
函式呼叫可能帶有參數,因此生成的組譯必須確保被呼叫函式的函式訊框在跳過去之前已設定好。舉例來說,若是平台的呼叫棧除了返回位置以外還有函式參數,編譯器需要輸出調整呼叫棧的指令。在這類平台上,考慮代碼:
function foo(data1, data2) a(data1) return b(data2)
其中 data1
、data2
是參數。編譯器會把這個代碼翻譯成以下組譯:
foo: mov reg,[sp+data1] ; 透过栈指针(sp)取得 data1 并放到暂用暂存器。 push reg ; 将 data1 放到栈上以便 a 使用。 call a ; a 使用 data1。 pop ; 把 data1 從栈上拿掉。 mov reg,[sp+data2] ; 透过栈指針(sp)取得 data2 並放到暂用暂存器。 push reg ; 将 data2 放到栈上以便 b 使用。 call b ; b 使用 data2。 pop ; 把 data2 從栈上拿掉。 ret
尾部呼叫最佳化會將代碼變成:
foo: mov reg,[sp+data1] ; 透过栈指针(sp)取得 data1 并放到暂用暂存器。 push reg ; 将 data1 放到栈上以便 a 使用。 call a ; a 使用 data1。 pop ; 把 data1 從栈上拿掉。 mov reg,[sp+data2] ; 透过栈指針(sp)取得 data2 並放到暂用暂存器。 mov [sp+data1],reg ; 把 data2 放到 b 预期的位置。 jmp b ; b 使用 data2 並返回到调用 foo 的函数。
更改後的代碼不管在執行速度或是棧空間的使用上的效能都比較好。
透過彈跳床
由於很多 Scheme 的編譯器使用C語言作為中間目標語言,問題可轉化為如何在 C 里在不讓棧向上增長的前提下實現尾端遞迴(假設 C 的編譯器不最佳化尾部呼叫)。很多實現透過一種叫做彈跳床(trampoline)的裝置,也就是一塊不斷進行函式呼叫的代碼。所有函式代碼的載入過程都透過這個彈跳床。當一個函式需要呼叫另一個函式時,它不是直接呼叫該函式,而是將該函式的位置、該呼叫使用的參數等資訊傳遞給彈跳床,讓愛插手的彈跳床去代為執行。這樣就可以確保 C 的棧不會向上增長並且可以讓迭代無限地繼續。
用 Groovy、Visual Basic .NET、C# 等等支援高階函式的語言實現彈跳床是可能的[9]。
對所有函式呼叫使用彈跳床,相比常規的C呼叫有著高昂的開銷,所以至少有一個Scheme編譯器即Chicken,使用了首先由Henry Baker聽從Andrew Appel未發表建議而描述的一種技術[10]。在其中使用常規的C呼叫,但是在每次呼叫前檢查棧的大小。當棧達到它的最大允許大小的時候,在棧上的對象經由Cheney演算法而被垃圾回收,所有存活資料都將移動到分立的堆之內。隨後棧被回縮(彈出),而程式恢復到緊鄰垃圾回收之前儲存的狀態。Baker聲稱:「Appel的方法通過偶爾的跳下帝國大廈而避免了大量的小型蹦床彈跳」[10]。垃圾回收確保了相互尾遞迴可以無限的繼續。但是這種方法要求C函式呼叫都永不返回,因為不能保證它的呼叫者的堆疊框仍然存在;故而它涉及到對程式碼的更加戲劇性的內部重寫,即將其改為續體傳遞風格。
更多實例
通常被用於解釋遞迴的程式是計算階乘。以下計算階乘的 Scheme 程式不是尾端遞迴,而只是一般遞迴[11]:
(define (factorial n)
(if (= n 1)
1
(* n (factorial (- n 1)))))
因此,如果呼叫 factorial
時的參數 n
足夠大,這一程式會出現堆疊溢位。然而,如果將同一程式寫作尾端遞迴,按 Scheme 的標準將不會出現溢位[11]:
(define (factorial n)
(define (iter product counter)
(if (> counter n)
product
(iter (* counter product)
(+ counter 1))))
(iter 1 1))
在第2個程式中,注意 iter
函式直接返回其遞迴呼叫,而沒有對其進行運算。因此,這是一個尾端遞迴,這讓直譯器或編譯器將本來是
call factorial (3) call iter (3 1) call iter (2 3) call iter (1 6) call iter (0 6) return 6 return 6 return 6 return 6 return 6
的執行過程組合成在時間、空間上效能都較好的型態:
call factorial (3) call iter (3 1) 将参数变为 (2 3),跳至 "iter" 将参数变为 (1 6),跳至 "iter" 将参数变为 (0 6),跳至 "iter" return 6 return 6
因為在中間過程中重複使用 iter
的函式訊框,這種重組節省了空間。這也代表程式設計師不需要為了擔心棧空間或是堆空間用完。在一般的實現中,尾端遞迴的型態也比其他型態更快速,不過僅僅是常數倍數的差異(非指數差異)。
很多使用函式語言的程式設計師會為了使用這個最佳化將遞迴的代碼寫成為尾端遞迴的形式。這通常需要一個多出來代表 「搜集器」 的形參(上述例子的 product
參數)。在一些語言中的一些函式的實現中(像是過濾一個列的實現等等),如果要使用尾端遞迴則需要將本來沒有副作用的純函式覆寫成會更動其他參引的形式[來源請求]。
注釋與資料
註釋
- ^ 1.0 1.1 1.2 1.3 1.4 1.5 1.6 1.7 Nicholas C. Zakas. 第3章“Functions”第9节“Tail Call Optimization”. Understanding ECMAScript 6 [理解ES6] 1. 245 8th Street, San Francisco, CA 94103: No Starch Press. 2016: 61–64. ISBN 978-1-59327-757-4 (英語).
- ^ recursion - Stack memory usage for tail calls - Theoretical Computer Science. Cstheory.stackexchange.com. 2011-07-29 [2013-03-21]. (原始內容存檔於2012-07-11) (英語).
- ^ Revised^6 Report on the Algorithmic Language Scheme. R6rs.org. [2013-03-21]. (原始內容存檔於2013-05-06).
- ^ Revised^6 Report on the Algorithmic Language Scheme - Rationale. R6rs.org. [2013-03-21]. (原始內容存檔於2013-11-11).
- ^ Revised^6 Report on the Algorithmic Language Scheme. R6rs.org. [2013-03-21]. (原始內容存檔於2018-03-15).
- ^ Python並沒有最佳化尾遞迴呼叫功能(以實現真正的尾遞迴特性),這裡只是用Python的語法來類比描述尾遞迴的語法。參見Does Python optimize tail-reccursion? (頁面存檔備份,存於網際網路檔案館)
- ^ 見《電腦程式的構造和解釋》第1章第2節「Procedure and Their Computation」。
- ^ Contact details. goto. perldoc.perl.org. [2013-03-21]. (原始內容存檔於2013-03-28).
- ^ Samuel Jack, Bouncing on your tail (頁面存檔備份,存於網際網路檔案館). Functional Fun. April 9, 2008.
- ^ 10.0 10.1 Henry Baker, "CONS Should Not CONS Its Arguments, Part II: Cheney on the M.T.A." (頁面存檔備份,存於網際網路檔案館)
- ^ 11.0 11.1 見《電腦程式的構造和解釋》。[頁碼請求]
參照
- Harold Abelson, Gerald Jay Sussman, Julie Sussman. Structure and Interpretation of Computer Programs [電腦程式的構造和解釋]. Cambridge, MA: MIT Press. 1996 [2011]. ISBN 0-262-01153-0. (原始內容存檔於2018年3月9日) (英語).