1. 程式人生 > 實用技巧 >LeetCode–刪除連結串列的節點

LeetCode–刪除連結串列的節點

LeetCode–刪除連結串列的節點

部落格說明

文章所涉及的資料來自網際網路整理和個人總結,意在於個人學習和經驗彙總,如有什麼地方侵權,請聯絡本人刪除,謝謝!

介紹

劍指 Offer 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 即可。
  • 初始化: pre = head , cur = head.next 。
  • 定位節點: 當 cur 為空 或 cur 節點值等於 val 時跳出。
    • 儲存當前節點索引,即 pre = cur 。
    • 遍歷下一節點,即 cur = cur.next 。
  • 刪除節點: 若 cur 指向某節點,則執行 pre.next = cur.next 。(若 cur 指向 nullnull ,代表連結串列中不包含值為 val 的節點。
  • 返回值: 返回連結串列頭部節點 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) {
        if(head.val == val){
            return head.next;
        }
        ListNode pre = head, cur = head.next;
        while(cur != null && cur.val != val){
            pre = cur;
            cur = cur.next;
        }
        if(cur != null){
            pre.next = cur.next;
        }
        return head;
    }
}

感謝

Leetcode

以及勤勞的自己,個人部落格GitHub