题目:
给定单向链表的头结点和一个结点,定义一个函数在O(1)时间内删除该结点。
分析:
删除链表结点常规做法为:按照链表特点从头结点找到该节点前一个结点,通过移动指针将指定结点从链表摘除,时间为O(n);
更简单方式为:通过已知的要删除结点,将下一个结点的值复制到该结点,再移动指针从链表摘除下一个结点,模拟删除该阶段操作,时间为O(1);
链表操作要注意链表节点的定义方式;头结点的返回;删除时要注意是否为尾结点等细节。
实现:
<span style="font-family:Microsoft YaHei;font-size:14px;">public ListNode delete(ListNode head, ListNode toDelete) {
if (head == null || toDelete == null)
return head;
// 要删除头结点
if (head == toDelete) {
return head.next;
}
// 删除最后一个结点
if (toDelete.next == null) {
ListNode temp = head;
while (temp.next != toDelete) {
temp = temp.next;
}
temp.next = null;
} else {
toDelete.val = toDelete.next.val;
toDelete.next = toDelete.next.next;
}
return head;
}</span>