演算法-如何理解遞迴,寫好遞迴函式
不是每個程式設計師天生對遞迴理解深刻,剛入大一時候,當別人寫出第一個求最大公約數的遞迴函式時,對其多麼的驚歎,竟然可以不用迴圈,竟然程式碼可以這麼簡潔,確實遞迴在大多數情況下實現時候程式碼很短,大部分人也知道遞迴,也能基本看懂遞迴,可是卻經常不知道怎麼寫,或者寫出來的遞迴經常死迴圈,寫演算法往往也是學的是套路,只有極少數人是創造演算法的,大部分人是用演算法的,而遞迴是確實有套路可循的。
本文即從遞迴的扎馬步開始,從幾個簡單例子到通用套路,一步一步拆解遞迴
1 遞迴的三要素
寫遞迴,就是寫三要素的實現,三要素分別為函式,邊界,遞推公式,剛開始只要記住要這麼寫,寫幾個演算法之後,就能慢慢明白為什麼要這樣搞。
1.1 遞迴首要元素-函式
明確你的函式是幹什麼用的,函式的入參應該是什麼,返回值是什麼,這三個問題,先從函式是幹什麼用的開始,你可以定義一個函式f()
假設已經實現了每一步遞迴的實現,再去明確這個實現 到底做了什麼,入參至少要什麼,返回值和引數返回可以理解為是一個東西,都是為了返回給上層呼叫或者全域性的一個數據,想清楚函式的三個要素,那你的函式就定義好了。
1.2 遞迴邊界、跳出遞迴
同樣,先這樣去做,再去想為什麼,這一步要判斷的就是函式的入參,入參的null
,入參為初始值,比如斐波那契數列的前1位或者2位,開始時候可能不一定想的完全,那沒關係,下面的一步還會繼續完善,所以我這裡舉得例子是斐波那契的前1或
if (n == 1)
return 1;
if (n == 2)
return 1;
想到的不一定完全對,或者那麼地很優雅, 沒關係,只要想到要考慮邊界就可以了。下面就是想邊界的意義是什麼?有兩點,其一,異常值邊界,其二遞迴結束判斷,比如此題中的n < 0 怎麼辦,和 n == 1
和 n == 2
就分別對應前面說的,當然這兩點可能考慮不那麼完全,假設你只考慮了像前面程式碼中的,或者寫邊界時候發現寫的多了,或者冗餘了,這樣不影響程式的結果,那麼寫完遞推公式,我們再來回顧邊界問題。
1.3 遞推公式
這個就要先談意義,再談實現了,意義在於逐漸減少演算法的規模,或者定義一種方式讓輸入的值儘可能地靠近臨界值,也就是找一個關係f(n)
與 f(n-x)
序列的關係,f(n)
代表要解決的問題規模,f(n-x)
比n小的問題規模的函式值,這是遞迴函式中的關鍵一步,沒有遞推公式,就沒有遞迴。例如斐波那契序列,中的遞推公式是f(n)=f(n-1) + f(n-2)
我們來觀察這個公式,發現其第n
於 n-1
和 n-2
有關係,所以我們來看,如果輸入任何整數,那麼n-1,n-2
可能取值是負數,0
,1+
,可以看到邊界0
和負數沒有考慮在內,所以,這時回顧前面1.2 的遞迴,我們來補充一下邊界後得到:
if (n <= 2)
return 1;
2 遞迴的案例
下面通過三個簡單例子,我們來練習使用遞迴,分別是青蛙跳臺階問題,等同於斐波那契序列,遞迴方式反轉單鏈表,遞迴遍歷樹,以及針對三個
2.1 青蛙跳臺階問題
第一定義函式,明確函式只有一個唯一的輸入值n
,第二找到遞迴結束條件或者遞迴邊界,可以發現當臺階是1或者2時候最容易得到,至於遞推式,可以發現青蛙在每次跳的時候有兩種跳法,那青蛙怎麼到達第n
個臺階,就是有兩種跳法,分別對應f(n-1)
和 f(n-2)
,所以遞迴式就是f(n)=f(n-1)+f(n-2)
,那麼整個演算法就是如下:
//一隻青蛙一次可以跳上1級臺階,也可以跳上2級。求該青蛙跳上一個n級的臺階總共有多少種跳法。
//1。定義函式
public int f2(int n) {
//2.確定邊界
if (n == 1)
return 1;
if (n == 2)
return 2;
//3.確定遞迴式
return f2(n-1) + f2(n-2);
}
繼續檢查邊界,發現n如果小於1,就會陷入死迴圈,那麼程式可以改成這樣:
if (n == 1)
return 1;
if (n == 2)
return 2;
if (n < 1)
return 0;
//當然簡單寫,可以這樣搞
if (n < 1)
return 0;
if (n <= 2)
return n;
2.2 遞迴方式反轉單鏈表
單鏈表的反轉,一般考慮到是雙指標反轉,當然遞迴寫也可以,同樣,首先定義函式,發現函式只有一個入參即節點node
這個node
在根節點或者任意中間節點都適用,其二確定邊界,在反轉單鏈表時候,可能會漏了node.next
的邊界,此時兩種方式,1,冗餘寫,只要你考慮到了,這可能是邊界,你多寫了絕對不會錯,甚至,你可以多寫兩到三步也完全沒問題,2,少寫的話,就寫完遞迴方式再來檢查,比如反轉單鏈表這個題,你會看到如果node.next
為空,那麼node.next.next
就會報空指標問題,一般寫完遞迴式後最好回頭檢查一下邊界,可以查缺補漏,去冗餘或者補條件。
此題的核心點是解開鏈的遞迴式,就是
Node last = f3(node.next); //假設當前節點的下一節點後的連結串列已經反轉
node.next.next = node; //當前節點指向的節點指向當前節點
node.next = null ;//原來當前節點是指向下一節點的,解開當前節點,並把節點指向空
//此處解釋,為什麼指向空,首先可以將node節點理解為第一個節點,那麼第一節點反轉後,就是最後一個節點,則指向是null,否則它還是指向2,就有問題喲
//那麼如果不是第一個節點呢?這個指標是怎麼指向的
舉個例子,假設,單鏈表是1,2,3,4,5
那麼遞迴的過程如下圖:
看圖,可以發現每一步的當前節點,放入反轉連結串列後,都是最後一個,那它必然指向null
這樣懂了把!
class Node{
int data;
Node next;
}
public Node f3(Node node) {
//2.確定返回邊界
if (node == null || node.next == null)
return node;
//3.拿到遞迴推導
Node last = f3(node.next);
node.next.next = node;
node.next = null ;//這個的作用是什麼?,解開死迴圈,最後是有A->B,B->A
return last;
}
2.3 遞迴遍歷樹
遞迴遍歷樹也是最簡單的,假設你之前沒有看過遍歷的程式碼,那麼從零來開始考慮這個問題,首先定義函式,確認入參和單鏈表反轉類似,只需要一個TreeNode
節點,然後考慮邊界為null
,和不為null
,你首先想到是不是這樣?
if (node == null)
return ;
if (node.left == null && node.right == null) {
System.out.pritln(node.val);
return ;
}
現在看起來是有點冗餘,但是假設你並不知道,那麼接下來下遞迴式,以先序為例
//首先節點本身
System.out.println(node.val);
//然後節點左
preOrder(node.left);
//然後節點右
preOrder(node.right);
就這樣完了,然後回顧前面的邊界問題,只有上面的程式碼兩行,可以看到在節點為null
的時候,就直接return
了,不用考慮子節點,位元組點的邊界在父節點的邊界中已經考慮到了,當然寫了這條邊界完全不影響程式執行哦,所以最終的前中後序遍歷如下程式碼:
//二叉樹前序遍歷
public static void preOrder(TreeNode node) {
if (node == null)
return;
System.out.println(node.val);
preOrder(node.left);
preOrder(node.right);
}
//二叉樹中序遍歷
public static void inOrder(TreeNode node) {
if (node == null)
return;
preOrder(node.left);
System.out.println(node.val);
preOrder(node.right);
}
//二叉樹後序遍歷
public static void postOrder(TreeNode node) {
if (node == null)
return;
preOrder(node.left);
preOrder(node.right);
System.out.println(node.val);
}
2.4 通過一個序列構造二叉樹
下面,我們補一個遞迴演算法題,輸入一個二叉樹序列,來還原構造二叉樹,順便測試一下前面的遍歷樹的程式碼,同樣熟悉了遞迴的套路後,我們直接來寫程式碼
//1.定義函式確認,只需要一個引數,即剩餘序列
public static TreeNode createBinaryTree(LinkedList<Integer> inputList) {
//定義一個空的樹節點,此處為了整齊劃一,在邊界和遞迴體裡面都可以用,所以寫在第一行
TreeNode node = null;
//2.邊界
if (inputList == null || inputList.isEmpty())
return node;
//3.主要遞迴體,從連結串列中刪除並取第一個元素,構建好左右節點,最後返回當前節點
Integer data = inputList.removeFirst();
//data,主要是異常值判斷,前面已經判斷過連結串列為空了
if (data != null) {
node = new TreeNode(data);
node.left = createBinaryTree(inputList);
node.right = createBinaryTree(inputList);
}
return node;
}
public static void main(String[] args) {
//前序遍歷序列
LinkedList<Integer> inputList = new LinkedList<Integer>(Arrays.asList(new Integer[]{3,2,9,null,null,10,null,null,8,null,4}));
TreeNode node = createBinaryTree(inputList);
//前序遍歷
System.out.println("前序遍歷:");
preOrder(node);
}
3. 總結
如何寫好遞迴,就三步,首先確認函式的輸入值,返回值,即函式本身要做什麼功能。其次,判斷邊界,將可以想到的邊界都寫一下。最後寫遞迴體,包括函式返回值,然後回去檢查邊界,對邊界增刪改查。
ps: 更多的情況下,只是沒想好演算法是怎麼樣,如果想好了,能夠用模擬法,把整個圖畫出來,寫程式碼就參考本文,即可一氣呵成。。。
吳邪,小三爺,混跡於後臺,大資料,人工智慧領域的小菜鳥。
更多請關注