LeetCode 203 Remove Linked List Elements
====================================
Remove all elements from a linked list of integers that have value val.
Example
Given: 1 --> 2 --> 6 --> 3 --> 4 --> 5 --> 6, val = 6
Return: 1 --> 2 --> 3 --> 4 --> 5
對于鏈表的刪除添加,一定要注意頭節(jié)點(diǎn)的操作,需要與其它節(jié)點(diǎn)分開處理。引入setinel作為偽頭節(jié)點(diǎn),可以避免分開處理簡化代碼。
對于刪除添加操作,一般的思路都是:
定義curr指針指向當(dāng)前結(jié)點(diǎn),pre指針指向curr的上一個節(jié)點(diǎn)。本題可以用類似思路,遍歷所有節(jié)點(diǎn),同時保留每個節(jié)點(diǎn)的上一個節(jié)點(diǎn),當(dāng)遇到節(jié)點(diǎn)值是val時,刪除該節(jié)點(diǎn)。
代碼:
public class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode setinel = new ListNode(0);
setinel.next = head;
ListNode pre = setinel;
ListNode curr = setinel.next;
while (curr != null) {
if (curr.val == val) {
pre.next = curr.next;
} else {
pre = curr;
}
curr = curr.next;
}
return setinel.next;
}
}
本題還可以直接先將從頭部開始的val值得節(jié)點(diǎn)去除,這樣就不需要考慮頭節(jié)點(diǎn)的特殊情況,也就不需要額外的偽頭節(jié)點(diǎn)了。
while (head != null && head.val == val) {
head = head.next;
}