删除链表中等于给定值 key 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
且只遍历单链表一遍!!!
先看一下我的思维逻辑图:
先定义一个cur:当前所要删除的节点
再定义一个prev:前驱
i)当cur不等于null时,判断是否是key值,如果等于,则让prev直接指向cur的下一个值,跳过cur即删除了与key相等的cur。
ii)如果不等,则让cur和prev继续往下走,重复 i) 步骤
iii)如果头节点就是所要删除的key值的的话,直接跳过,让下一个节点变成新的头节点即可。
像我画的图就是不带头节点的单链表,那么程序代码就如下所示:
class Solution {
public ListNode removeElements(ListNode head, int val) {
if(head == null) {
return null;
}
ListNode prev = head;
ListNode cur = head.next;
while(cur != null){
if(cur.val == val ){
prev.next = cur.next;
cur = cur.next ;
}else{
prev = cur;
cur = cur.next;
}
}
if(head.val == val) {
head = head.next;
}
return head;
}
}
附带头结点的单链表的程序:
class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode prev = new ListNode(-1);
prev.next = head;
ListNode cur = prev;
while(cur.next != null){
if(cur.next.val == val ){
cur.next = cur.next.next;
}else{
cur = cur.next;
}
}
return prev.next;
}
}