[[403833]]

本文轉載自微信公眾號「程序員小熊」，作者程序員小熊。轉載本文請聯系程序員小熊公眾號。

前言

大家好，我是來自華為的「程序員小熊」。相信絕大部分童鞋都知道，在處理與「鏈表」相關問題時，常用的解題套路主要包括「雙指針」、「迭代」和「虛擬頭節(jié)點」等等。

今天「小熊」主要介紹采用「遞歸」的策略，秒殺「鏈表」相關問題，使得代碼更「優(yōu)雅」，并以兩道常見的面試題作為例題來講解，供大家參考，希望對大家有所幫助。

鏈表與遞歸

鏈表具有天然的遞歸性，一個鏈表可以看出頭節(jié)點后面掛接一個更短的鏈表，這個更短的鏈表是以原鏈表的頭節(jié)點的下一節(jié)點為頭節(jié)點，依次內推，直到最后的更短的鏈表為空，空本身也是一個鏈表(最基礎的)。

以單鏈表 1->2->3->null 為例子，如下圖示：

原鏈表

將原鏈表看出頭節(jié)點 1 后掛接一個更短的鏈表

頭節(jié)點+更短鏈表

繼續(xù)拆解，直到無法拆解

更更短鏈表

更更更短鏈表

有了這樣的思考，很多「鏈表」相關問題，都可以采用「遞歸」的思路來解答。

劍指 Offer 24. 反轉鏈表

定義一個函數，輸入一個鏈表的頭節(jié)點，反轉該鏈表并輸出反轉后鏈表的頭節(jié)點。 
  
 
示例: 
 
輸入: 1->2->3->4->5->NULL 
輸出: 5->4->3->2->1->NULL 
  
 
限制： 
0 <= 節(jié)點個數 <= 5000 

解題思路

要反轉鏈表，即將原鏈表的頭節(jié)點變?yōu)樾骆湵淼奈补?jié)點，原鏈表的尾節(jié)點變?yōu)樾骆湵淼念^節(jié)點。如下圖示：

反轉之前：

原鏈表

反轉之后：

新鏈表

主要策略主要有：1、直接修改鏈表的值，如上圖中的原鏈表圖所示，將原鏈表值 1 的頭節(jié)點改為原鏈表尾節(jié)點的值，依次類推;2、讓遍歷整個鏈表，讓鏈表的尾節(jié)點指向其前一個節(jié)點，依次類推。

雖然這兩個策略都可行，不過在面試中通常要求采用「策略2」。

由上面的「遞歸與鏈表」可知，本題也可以采用「遞歸法」去求解。

具體如何通過「遞歸」去解答呢?見下面例子。

「舉例」

鏈表 1->2->3->null 為例子，如下圖示。

示例

不斷遍歷找到原鏈表為尾節(jié)點，即新鏈表的頭節(jié)點。

原鏈表尾節(jié)點

然后讓尾節(jié)點指向其前驅節(jié)點，依次類推。

遞歸反轉

詳細步驟，如下動圖示：

遞歸反轉單鏈表

Show me the Code

「C」

struct ListNode* reverseList(struct ListNode* head){ 
    /* 遞歸終止條件 */ 
    if (head == NULL || head->next == NULL) { 
        return head; 
    } 
 
    /* 反轉當前所在的鏈表節(jié)點 */ 
    struct ListNode* newHead = reverseList(head->next); 
 
    /* 由原鏈表的尾節(jié)點挨個指向其前驅節(jié)點 */ 
    head->next->next = head; 
 
    /* 防止成環(huán) */ 
    head->next = NULL; 
 
    /* 返回新的鏈表頭節(jié)點 */ 
    return newHead; 
} 

「java」

ListNode reverseList(ListNode head) { 
    if (head == null || head.next == null) { 
        return head; 
    } 
 
    ListNode node = reverseList(head.next); 
    head.next.next = head; 
    head.next = null; 
 
    return node; 
} 

當然本題也可以采用「迭代」的方法去做，其代碼(python 版)也很優(yōu)雅，具體如下：

Show me the Code

「python」

def reverseList(self, head: ListNode) -> ListNode: 
    cur, pre = head, None 
    while cur: 
        cur.next, pre, cur = pre, cur, cur.next 
         
    return pre  

「復雜度分析」

時間復雜度：「O(n)」，n 是鏈表的長度。對鏈表的每個節(jié)點都進行了反轉操作。

空間復雜度：「O(n)」，n 是鏈表的長度。遞歸調用的棧空間，最多為 n 層。

203. 移除鏈表元素

給你一個鏈表的頭節(jié)點 head 和一個整數 val ，請你刪除鏈表中所有滿足  
 
Node.val == val 的節(jié)點，并返回 新的頭節(jié)點 。 
 
示例 1： 
 
輸入：head = [1,2,6,3,4,5,6], val = 6 
輸出：[1,2,3,4,5] 
 
示例 2： 
 
輸入：head = [], val = 1 
輸出：[] 
 
示例 3： 
 
輸入：head = [7,7,7,7], val = 7 
輸出：[] 

解題思路

要移除鏈表中給定值的節(jié)點，一般策略是「找到給點值的節(jié)點的前驅節(jié)點，然后讓其指向給定值的節(jié)點的后繼節(jié)點」。

例如要刪除鏈表 1->2->3->null 中，節(jié)點值為 3 的節(jié)點，就得先找到其前驅節(jié)點(值為 2 的節(jié)點)，如下圖示：

刪除給定值的節(jié)點

由上面的「遞歸與鏈表」可知，本題同樣也可以采用「遞歸法」去求解，不斷刪除更短鏈表中給定值的節(jié)點，然后再將處理后的更短的鏈表，掛接在其前驅節(jié)點后。

「注意」最后要判斷原鏈表的頭節(jié)點是否是待移除的節(jié)點。

「舉例」

以鏈表 1->2->3->null 為例子，移除鏈表中給定值的節(jié)點的過程，如下動圖示。

示例動圖

Show me the Code

「C++」

ListNode* removeElements(ListNode* head, int val) { 
    /* 遞歸終止條件 */ 
    if(head == NULL) { 
        return NULL; 
    } 
 
    /* 刪除鏈表中頭節(jié)點后值為 val 的元素的節(jié)點 */ 
    head->next=removeElements(head->next,val); 
 
    /* 判斷頭節(jié)點是否為值為 val 的節(jié)點，再返回*/ 
    return head->val==val ? head->next : head; 
} 

「Golang」

func removeElements(head *ListNode, val int) *ListNode { 
    if head == nil { 
        return head 
    } 
 
    head.Next = removeElements(head.Next, val) 
    if head.Val == val { 
        return head.Next 
    } 
 
    return head 
} 

「復雜度分析」

時間復雜度：「O(n)」，n 是鏈表的長度。遞歸需要遍歷鏈表一次。

空間復雜度：「O(n)」，n 是鏈表的長度。遞歸調用棧，最多不會超過 n 層。