给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。返回删除后的链表的头节点。如输入: head = [4,5,1,9], val = 5 输出: [4,1,9]。
要删除单链表上的某一个固定节点,则首先需要找到该节点,然后将该节点的后置节点挂到该节点的前节点上即可。因此,只需要定义两个变量,分别表示节点的前置节点和后置节点,对节点进行遍历,当满足条件时,则break掉中断循环。
package learnproject.offer;
/*
* 18 删除链表的节点
*/
public class Demo18 {
/*
* 1.要删除链表中的某一个元素,则需要找到值为val的节点,
* 然后将该节点的后一个节点挂到该节点的前一个节点即可
* 2.当head节点为空时,返回null
* 3.如果head节点的val和输入值相同,则将链表的head节点删除,返回以head.next节点为头节点的链表
* 4.如果头节点不是要删除的节点,定义两个临时变量next,pre,分别表示当前节点和上一个节点进行while循环
* 5.循环过程中如果发现值为val的节点,则将next.next赋值给pre.next
* 6.最后返回以head为头节点的链表
*/
public ListNode deleteNode(ListNode head, int val) {
if(head == null) {
return head;
}
int value = head.val;
if(val == value) {
return head.next;
}
ListNode next = head.next;
ListNode pre = head;
while(next!=null) {
value = next.val;
if(val == value) {
pre.next = next.next;
break;
}else {
pre = next;
next = next.next;
}
}
return head;
}
public static void main(String args[]) {
Demo18 demo = new Demo18();
ListNode node1 = new ListNode(4);
ListNode node2 = new ListNode(5);
ListNode node3 = new ListNode(1);
ListNode node4 = new ListNode(9);
node1.next = node2;
node2.next = node3;
node3.next = node4;
ListNode newNode = demo.deleteNode(node1, 5);
while(newNode!=null) {
System.out.println(newNode.val);
newNode = newNode.next;
}
}
}