题目:给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
方法一:不使用虚拟头结点
class Solution1 {
public ListNode removeElements(ListNode head, int val) {
// 情况1:空链表
if(head == null)
return null;
// 情况2:链表不为空
// 第一步:头结点需要删除
while(head.val == val) {
if(head.next == null)
return null;
else {
head = head.next;
}
}
// 第二步:头结点不需要删除
ListNode prev = head;
while(prev.next != null) {
if(prev.next.val == val) {
prev.next = prev.next.next;
} else {
prev = prev.next;
}
}
return head;
}
}
方法二:使用虚拟头结点
// 使用虚拟头结点
// 若有虚拟头结点,就不用判断第一个节点是否需要删除,
// 只用按照方法一中的第二步进行处理即可。
class Solution2 {
public ListNode removeElements(ListNode head, int val) {
ListNode dummyHead = new ListNode(-1); //向虚拟头结点中传入任意数
dummyHead.next = head;
// 第二步:头结点不需要删除
ListNode prev = dummyHead;
while(prev.next != null) {
if(prev.next.val == val) {
prev.next = prev.next.next;
} else {
prev = prev.next;
}
}
return dummyHead.next;
}
}
方法三:使用递归的思想
class Solution3 {
public ListNode removeElements(ListNode head, int val) {
if(head == null) {
return null;
}
ListNode cur = head.next;
if(head.val == val) {
head = removeElements(cur, val);
return head;
} else {
head.next = removeElements(cur, val);
return head;
}
}
}
还可写为:
class Solution3 {
public ListNode removeElements(ListNode head, int val) {
if(head == null) {
return null;
}
head.next = removeElements(head.next, val);
return head.val == val ? head.next : head;
}
}