LeetCode 203:移除鏈表元素 Remove LinkedList Elements

刪除鏈表中等於給定值 val 的全部節點。java

Remove all elements from a linked list of integers that have value val.python

示例:函數

輸入: 1->2->6->3->4->5->6, val = 6
輸出: 1->2->3->4->5

解題思路:

兩種方法,一種是迭代法,從第一個節點開始,遇到值相同的節點就將其刪除。鏈表的刪除操做是直接將刪除節點的前一個節點指向刪除節點的後一個節點便可。學習

第二種方法是遞歸,用遞歸從後向前遇到相同節點直接指向該節點的下一個節點的地址便可指針

迭代法:

因爲鏈表刪除操做的特殊性,若是要刪除某個節點,必需要知道該刪除節點的前一個節點地址纔可完成刪除操做。因此若是是從第一個節點開始判斷,就要考慮到第一個節點是否爲空節點、第一個節點是否就是該刪除的的節點,刪除頭節點和非頭節點的操做不同,應單獨實現刪除操做。code

若是原鏈表是這種形式:1->1->1->2 val=1 刪除頭節點後第二個節點置爲頭節點,可是第二個節點做爲新的頭節點依然須要刪除,因此對頭節點的操做應該是一個迭代過程。遞歸

另一種方法就是新建一個虛擬節點,該虛擬節點下一個節點指向原鏈表頭節點。這就無需考慮頭節點是否爲空、是否爲待刪除節點。如原鏈表爲:1->1->1->2 val=1 ,新建一個虛擬節點 -1 做爲頭節點:-1->1->1->1->2 val=1 ,這時只需正常迭代刪除便可,惟一要注意是返回節點不能是 head ,由於原鏈表的頭節點若是是待刪除節點,此時 head 節點已被刪除並做爲單獨隔離出來的節點,並不是鏈表內的一個節點。element

Java:rem

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode newHead=new ListNode(-1);//新建虛擬節點
        newHead.next=head;//虛擬節點做爲原鏈表的頭節點
        ListNode cur = newHead;//遍歷節點的指針

        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;//刪除操做
            } else {
                cur = cur.next;
            }
        }
        return newHead.next;//返回的頭節點應當是虛擬節點的下一個節點
    }
}

Python3:io

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        newHead = ListNode(-1)
        newHead.next = head
        cur = newHead
        while cur.next:
            if cur.next.val == val:
                cur.next = cur.next.next
            else:
                cur = cur.next
        return newHead.next

遞歸法:

遞歸方法解該題很簡單,基線條件是遇到空節點(最後一個節點),遞歸時只需將傳遞參數節點的下一個節點做爲新的參數傳給遞歸函數便可:

如原鏈表爲: 1->2->6->3->4->5->6, val = 6 遞歸到空節點時最後一個遞歸函數返回null: 1->2->6->3->4->5->6->null 回到上一層遞歸函數內此時 head.next 獲得返回節點 null 判斷head爲 6 刪除,返回 head.next :1->2->6->3->4->5->null 回到上一層遞歸函數此時 head.next 獲得上一層返回節點依然爲null 判斷head爲 5 不刪除,返回 head 爲5:1->2->6->3->4->5->null 回到上一層遞歸函數此時 head.next 獲得上一層返回節點 5 判斷head爲 4 不刪除,返回 head 爲4:1->2->6->3->4->5->null 回到上一層遞歸函數此時 head.next 獲得上一層返回節點 4 ..............

直到回到第一個遞歸函數爲止返回頭節點結束。

Java:

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return null;//基線條件
        head.next = removeElements(head.next, val);
        if (head.val == val) {
            return head.next;
        } else {
            return head;
        }
    }
}

Python3:

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        if not head: return None
        head.next = self.removeElements(head.next, val)
        if head.val == val:
            return head.next
        else:
            return head

歡迎關注公衆號一塊兒學習:愛寫Bug

愛寫Bug.png

相關文章
相關標籤/搜索