leetcode筆記_連結串列的遍歷與刪除(虛擬頭結點的使用)
阿新 • • 發佈:2018-11-10
203. 移除連結串列元素
刪除連結串列中等於給定值 val 的所有節點。
示例:
輸入: 1->2->6->3->4->5->6, val = 6 輸出: 1->2->3->4->5
思路:連結串列的增加和刪除操作,一定要通過待處理節點的前一個節點來實現,為了解決頭結點沒有前繼節點的特殊性,需要給連結串列新增一個虛擬頭結點。新建指標一個用於索引的指標prev, 指向待處理節點的前繼節點,則待處理節點為prev.next. 初始化prev則指向連結串列的虛擬頭結點。
解法1. 不使用虛擬頭結點,需要考慮頭結點的特殊處理,比較複雜
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { val = x; } * } */ class Solution { public ListNode removeElements(ListNode head, int val) { //無虛擬頭結點,需要對頭結點做特殊處理 //將連結串列開頭連續的值為val的節點都刪除 while(head!=null && head.val == val){ head=head.next; } if(head==null){ return null; } //對頭結點的值不為val的連結串列繼續做刪除操作 ListNode cur = head; while(cur.next != null){ if(cur.next.val == val){ cur.next = cur.next.next; }else{ cur = cur.next; } } return head; } }
解法2. 使用虛擬頭結點,程式碼簡介,思路清晰
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { val = x; } * } */ class Solution { public ListNode removeElements(ListNode head, int val) { //建立虛擬頭結點 ListNode dummyHead = new ListNode(-1); dummyHead.next = head; //建立遍歷連結串列所用的指標,prev指向的是帶處理元素的前一個元素,待處理元素為prev.next ListNode prev = dummyHead; //對連結串列進行遍歷,注意連結串列為空的判斷 while(prev.next!=null){ if(prev.next.val == val){ prev.next = prev.next.next; }else{ prev = prev.next; } } return dummyHead.next; } }