一步一步寫演算法(之遞迴和堆疊)
看過我前面部落格的朋友都清楚,函式呼叫主要依靠ebp和esp的堆疊互動來實現的。那麼遞迴呢,最主要的特色就是函式自己呼叫自己。如果一個函式呼叫的是自己本身,那麼這個函式就是遞迴函式。
我們可以看一下普通函式的呼叫怎麼樣的。試想如果函式A呼叫了函式B,函式B又呼叫了函式C,那麼在堆疊中的資料是怎麼儲存的呢?
函式A ^
函式B | (地址遞減)
函式C | 如果是遞迴函式呢,舉一個簡單的遞迴函式為例:
int iterate(int value)
{
if(value == 1)
return 1;
return value + iterate(value -1);
} 下面我們使用一個函式進行呼叫,看看會發生什麼情況?
void process()
{
int value = iterate(6);
} 看看此時記憶體堆疊是什麼樣的?
iterate(int 1) line 96
iterate(int 2) line 97 + 12 bytes
iterate(int 3) line 97 + 12 bytes
iterate(int 4) line 97 + 12 bytes
iterate(int 5) line 97 + 12 bytes
iterate(int 6) line 97 + 12 bytes
process() line 102 + 7 bytes
main() line 108
mainCRTStartup() line 206 + 25 bytes
KERNEL32! 7c817067()
大家也看到了上面的程式碼,遞迴函式和普通的函式也沒有什麼差別。除了自己呼叫本身之外,他就是一個普通的函式。那麼這個函式遞迴到什麼時候返回呢?這就是遞迴函式的關鍵了。我們看到iterate函式到1就停止了,所以上面的堆疊在(value == 1)即return。所以一個遞迴函式最關鍵的部分就是兩點:(1)遞迴策略;(2)函數出口。
看到這裡,大家可能感到遞迴函式不過如此,事實上也是這樣。但是,還有一點大家需要牢記在心,遞迴的深度是我們必須考慮的一個問題。只有遞迴深度在一個可控的範圍內,那麼整個遞迴過程都是可控的。那什麼時候不可控呢?那就是遞迴深度超過了一定的數字?這個數字和具體的執行緒堆疊長度有關?等到堆疊溢位了,那麼獲得的資料已經失去了真實性,所以也就沒有意義了。
我們把上面的問題推廣一下,如何用自己定義的堆疊模擬上面的遞迴呼叫呢?這樣既能滿足遞迴的屬性,又能確保函式深度可控。
大家可以先寫一下自己的方案,下面只是我個人的一個思路。
int iterate(int value)
{
int count = 0;
int number =0;
push(value);
while(-1 != (number = pop()))
{
if(1 != number)
push(number -1);
count += number;
}
return count;
}