题目描述:
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。
返回删除后的链表的头节点。
注意:此题对比原题有改动
示例 1:
输入: head = [4,5,1,9], val = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.
示例 2:
输入: head = [4,5,1,9], val = 1
输出: [4,5,9]
解释: 给定你链表中值为 1 的第三个节点,那么在调用了你的函数之后,该链表应变为 4 -> 5 -> 9.
解题思路:顺着题目写就行,对于没学过数据结构的应该就是比较困难了,但是学过数据结构就很轻松了,注意一点null就行。但是内存占用并不理想。
代码:
public class Q18 {
public static void main(String[] args) {
ListNode head = new ListNode(4);
ListNode head1 = new ListNode(5);
ListNode head2 = new ListNode(1);
ListNode head3 = new ListNode(9);
head.next = head1;
head1.next = head2;
head2.next = head3;
ListNode node = head;
// while(node != null) {
// System.out.println(node.val);
// node = node.next;
// }
head = deleteNode(head,4);
node = head;
while(node != null) {
System.out.println(node.val);
node = node.next;
}
}
public static ListNode deleteNode(ListNode head, int val) {
if(head != null) {
if(val == head.val) {
return head.next;
}else {
ListNode node = head;
while(node.next != null) {
if(val == node.next.val) {
node.next = node.next.next;
return head;
}
node = node.next;
}
return head;
}
}
return null;
}
}
class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}