题目:在O(1)时间内删除链表节点。给定单向链表的头指针和一个节点指针,定义一个函数deleteNode()在O(1)时间内删除该节点。链表节点定义如下:
class ListNode {
int value;
ListNode next;
public ListNode(int value) {
this.value = value;
this.next = null;
}
// getter and setter ...
}
在单向链表中删除一个节点,常规的做法一般就是从链表的头节点开始遍历查找要删除的节点,并在链表中删除该节点。如下图中的(a)链表,我们若想删除节点[i],则需要从头节点[a]开始顺序遍历,得到节点[i]的上一个节点[h],将节点[h]的next指向节点[i]的下一个节点[j],如图中的(b),实现删除。单向链表中删除节点的常规做法需要O(n)的时间复杂度。
既然从头节点遍历的时间复杂度不满足要求,那我们能不能不遍历呢?
其实我们可以如图中的(c)那样,将节点[i]的下一节点[j]的值赋给节点[i],然后将节点[i]的next指向节点[j]的next,最后删除节点[j]。这种方法时间复杂度为O(1),满足题目要求。
需要注意两点:1.如果要删除的节点是链表的尾节点,那么依然需要从头节点开始遍历得到尾节点的上一节点;2.如果链表中只有头节点,并且要删除的节点就是头节点,此时删除后链表应该为null。
以下代码中deleteNode()方法传入链表头节点和一个要删除的节点(默认就是链表中的一个节点,没有考虑该节点不是链表中的节点的情况),返回删除节点后的链表。
class Solution {
public ListNode deleteNode(ListNode head, ListNode deleted) {
if (head == null || deleted == null) return null;
if (deleted.next != null) { // deleted的next不为空,说明要删除的节点不是尾节点
deleted.value = deleted.next.value;
deleted.next = deleted.next.next;
} else if (head == deleted) { // 链表只有一个头节点
return null;
} else { // 要删除的节点是尾节点(使用常规方法:顺序遍历删除)
ListNode pre = head, cur = head.next;
while (cur != deleted) {
pre = cur;
cur = cur.next;
}
pre.next = null;
}
return head;
}
}