問題:
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
大意:
移除鏈表中素有值為val的元素。
例子:
給出: 1 --> 2 --> 6 --> 3 --> 4 --> 5 --> 6, val = 6
返回: 1 --> 2 --> 3 --> 4 --> 5
思路:
這道題沒什么特殊的,遍歷鏈表遇到值跟val相同的就刪除節點就好,鏈表刪除節點的做法也比較固定了,只是要細心一點不要漏了什么特殊情況,比如刪除節點時是最后一個節點的處理、鏈表頭部連續出現要刪除的節點時的處理等等,另外因為最終要返回的是頭結點,所以這個節點要有一個變量去保留著最后返回。
我的做法是先把頭部的連續的目標節點都刪掉,然后遍歷鏈表,中間遇到了就刪,當然遇到時要判斷是不是最后一個節點了,是的話就next直接指向null了。在刪頭結點時,因為可能循環著遇到后面沒節點了的情況,比如 1 --> 1 這種全部要刪除的,所以在循環條件中利用 && 運算符的特性,先判斷節點存在,然后再判斷值,如果節點不存在,第一個判斷條件就無效,那么就不會處理第二個判斷條件,否則直接判斷值有可能在無節點時報錯。
代碼(Java):
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
public class Solution {
public ListNode removeElements(ListNode head, int val) {
if (head == null) return head;
ListNode result = head;
while (result != null && result.val == val) result = result.next;
while (head.next != null) {
if (head.next.val == val) {
if (head.next.next != null) head.next = head.next.next;
else head.next = null;
} else {
head = head.next;
}
}
return result;
}
}
合集:https://github.com/Cloudox/LeetCode-Record