1. 程式人生 > >LeetCode 203:移除連結串列元素 Remove LinkedList Elements

LeetCode 203:移除連結串列元素 Remove LinkedList Elements

刪除連結串列中等於給定值 val 的所有節點。

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

示例:

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

解題思路:

兩種方法,一種是迭代法,從第一個節點開始,遇到值相同的節點就將其刪除。連結串列的刪除操作是直接將刪除節點的前一個節點指向刪除節點的後一個節點即可。

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

迭代法:

由於連結串列刪除操作的特殊性,如果要刪除某個節點,必須要知道該刪除節點的前一個節點地址才可完成刪除操作。所以如果是從第一個節點開始判斷,就要考慮到第一個節點是否為空節點、第一個節點是否就是該刪除的的節點,刪除頭節點和非頭節點的操作不一樣,應單獨實現刪除操作。

如果原連結串列是這種形式:1->1->1->2 val=1 刪除頭節點後第二個節點置為頭節點,但是第二個節點作為新的頭節點依然需要刪除,所以對頭節點的操作應該是一個迭代過程。

另外一種方法就是新建一個虛擬節點,該虛擬節點下一個節點指向原連結串列頭節點。這就無需考慮頭節點是否為空、是否為待刪除節點。如原連結串列為:1->1->1->2

val=1 ,新建一個虛擬節點 -1 作為頭節點:-1->1->1->1->2 val=1 ,這時只需正常迭代刪除即可,唯一要注意是返回節點不能是 head ,因為原連結串列的頭節點如果是待刪除節點,此時 head 節點已被刪除並作為單獨隔離出來的節點,並非連結串列內的一個節點。

Java:

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:

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