[LeetCode]（面試題18）刪除鏈表的節點

題目

給定單向鏈表的頭指針和一個要刪除的節點的值，定義一個函數刪除該節點。

返回刪除後的鏈表的頭節點。

注意：此題對比原題有改動

示例 1:

輸入: head = [4,5,1,9], val = 5
輸出: [4,1,9]
解釋: 給定你鏈表中值爲 5 的第二個節點，那麼在調用了你的函數之後，該鏈表應變爲 4 -> 1 -> 9.

示例 2:

輸入: head = [4,5,1,9], val = 1
輸出: [4,5,9]
解釋: 給定你鏈表中值爲 1 的第三個節點，那麼在調用了你的函數之後，該鏈表應變爲 4 -> 5 -> 9.

說明：

• 題目保證鏈表中節點的值互不相同
• 若使用 C 或 C++ 語言，你不需要 free 或 delete 被刪除的節點

解題思路

算法步驟：
特例處理： 當應刪除頭節點 head 時，直接返回 head.next 即可。
初始化： cur = head。
定位節點： 當 cur.next 非空，並且 cur.next 的節點值不等於 val 時進入循環，跳到下一節點：cur = cur.next 。
當 cur.next 非空（即不是尾節點末尾時），表示找到了要刪除的節點，那麼就更改引用關係：執行 cur.next = cur.next.next。（若跳出循環時 cur.next = null，代表鏈表中不包含值爲 val 的節點。）
返回值： 返回鏈表頭部節點 head 即可。

複雜度分析：
時間複雜度：O(N)。
空間複雜度：O(1)。

注：也可以在鏈表前面加一個僞頭指針，這樣就不用再判斷原來的第一個頭節點是否爲要刪除的節點了。

代碼

解法一：不添加僞頭指針

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode deleteNode(ListNode head, int val) {
        if(head.val == val){
            return head.next;
        }
        ListNode cur = head;
        while(cur.next != null && cur.next.val != val){
            cur = cur.next;
        }
        if(cur.next != null){
            cur.next = cur.next.next;
        }
        return head;
    }
}

解法二：添加僞頭指針

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode deleteNode(ListNode head, int val) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        head = dummy;
        while(head.next != null && head.next.val != val){
            head = head.next;
        }
        if(head.next != null){
            head.next = head.next.next;
        }
        return dummy.next;
    }
}