LeetCode題解 83. 刪除排序連結串列中的重複元素

阿新 • • 發佈：2020-12-10

LeetCode題解 83. 刪除排序連結串列中的重複元素

文章目錄

LeetCode題解 83. 刪除排序連結串列中的重複元素

題目描述

給定一個排序連結串列，刪除所有重複的元素，使得每個元素只出現一次。

示例一:
輸入: 1->1->2
輸出: 1->2


示例二:
輸入: 1->1->2->3->3
輸出: 1->2->3

題解

迭代寫法

解題思路

1. 用cur表示當前節點，利用wile迴圈遍歷連結串列
2. 結束條件為: 遍歷完連結串列，即 cur.next 為空; 
             頭結點 head 可能為空, 則也需要判斷 cur 是否為空
3. 遍歷過程中：
	(1)如果該節點出元素與下一節點元素相同，
	   則刪除下一節點: cur.next = cur.next.next;
	(2)若不同，則繼續遍歷;

細節分析

1. 需要判斷 head 是否為空:
	(1) 可在開頭判斷 head 是否為空
	(2) 也可在 while 迴圈中判斷 cur 是否為空
2. 為什麼如果前後兩節點元素相同，刪除後節點，而不用 cur = cur.next 操作？
答：是因為刪除該節點的下一節點，該節點仍需要和新的下一節點進行比較

程式碼

java

class Solution {
    public ListNode deleteDuplicates(ListNode head) {
        ListNode cur = head;
        //cur != null 的作用是判斷 head 是否為 null 

        while (cur != null && cur.next != null) {
            if (cur.next.val == cur.val) {
                //刪除cur.next節點
                cur.next = cur.next.next;
            } else {
                cur = cur.next;
            }
        }
        return head;
    }
}

class Solution { 

public:
    ListNode* deleteDuplicates(ListNode* head) {
        ListNode* cur = head;
        while (cur != NULL && cur->next != NULL) {
            if (cur->val == cur->next->val) {
                cur->next = cur->next->next;
            } else {
                cur = cur->next;
            }
        }

        return head;
    }
};

python

class Solution:
    def deleteDuplicates(self, head: ListNode) -> ListNode:
        cur = head
        while cur and cur.next:
            if cur.val == cur.next.val:
                cur.next = cur.next.next
            else: cur = cur.next

        return head

遞迴寫法

解題思路

* 遞迴寫法的刪除節點操作是 通過返回下一節點 連線到上一節點，
* 來實現引數節點(即當前節點)的刪除

遞迴的三步法:
(引數 head 的含義為當前節點)

1. 終止條件: 遍歷完連結串列
	(1) head.next 為空
	(2) head 為空，即空連結串列
2. 終止處理: 返回當前節點
3. 提取重複邏輯:
	(1) 節點的"連線": 利用函式返回值作為下一節點
	(2) 節點的"刪除": 判斷是否當前節點元素與下一節點元素是否相同, 返回相應的節點

程式碼

java

class Solution {
    public ListNode deleteDuplicates(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }

        head.next = deleteDuplicates(head.next);

        return head.val == head.next.val ? head.next : head;
    }
}

class Solution {
public:
    ListNode* deleteDuplicates(ListNode* head) {
        if (head == NULL || head->next == NULL) {
            return head;
        }

        head->next = deleteDuplicates(head->next);

        return head->val == head->next->val ? head->next : head;
    }        
};

python

class Solution:
    def deleteDuplicates(self, head: ListNode) -> ListNode:
        if not head or not head.next: return head
        head.next = self.deleteDuplicates(head.next)

        return head.next if head.val == head.next.val else head