题目要求:
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。
返回删除后的链表的头节点。
public class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
时间复杂度O(n)解法:
1.若链表为空,直接返回空
2.若要删除头节点,直接返回第二个节点
3.然后遍历,条件是当前节点下一个节点的值不是要删除的元素且没有到达链表末尾
4.找到要删除的节点的前一个节点,删除。
class Solution {
public ListNode deleteNode(ListNode head, int val) {
if(head==null){return null;}
if(head.val==val){return head.next;}
ListNode rec=head;
while(rec.next.val!=val&&rec.next!=null){
//找到要删除节点的上一个值
rec=rec.next;
}
//判定不是因为遍历完链表都没找到结束循环
if(rec.next!=null) rec.next=rec.next.next;
return head;
}
}
要求时间复杂度为O(1)解法
解法在注释中
public ListNode deleteNode(ListNode head, ListNode val){
//链表为空,直接返回
if(head==null) return head;
//需要删除的节点后面还有节点时,先把下个节点的内容复制到需要删除的节点上覆盖原有的内容,再把下一个节点删除
if(val.next!=null) {
val.val= val.next.val;
val.next = val.next.next;
return head;
}
/*需要删除的节点后面没有节点时,即要删除的节点位于尾节点
两个情况:1.链表中只有这一个节点,此时删除后链表为null。*/
if(head==val) { //1.链表中只有这一个节点,且为被删除节点时,此时删除后链表为null。
return null;
}
//2.需要删除的节点前面还有节点。
ListNode cur =head;
while(cur.next!=null&&cur.next!=val) {
cur=cur.next;
}
//如果需要删除的节点不在链表里,则不删除
if(cur.next!=null) cur.next=cur.next.next;
return head;
}