题目
该系列文章题目和思路均参考自:《剑指Offer》- Page 119
解法
在单向链表中删除一个节点,常规的做法是从链表的头结点开始,顺序遍历查找要删除的节点,然后修改该节点前一个节点的next指针,指向要删除节点的后一个节点。这样可以安全删除链表中要删除的节点。但这种思路需要知道待删除节点的前一个节点才能进行,而且时间复杂度为O(n)。如下图中的(a)和(b)所示。
那是不是一定需要得到被删除的节点的前一个节点呢?答案是否定的。我们可以把待删除节点的下一个节点的内容复制到待删除节点中,然后删除待删除节点的下一个节点就可以实现在O(1)的时间复杂度内的节点删除操作,而且不需要知道待删除节点的前一个节点。
当然,这种思路还有一个问题,就是如果待删除的节点为尾节点怎么办?由于他没有下一个节点可以使用,所以,只能使用传统的方式,从链表的头结点开始遍历得到该节点的前序节点,然后将尾节点进行删除。
此外还有一点需要注意,如果链表本身只有一个节点,那么在删除掉这个头结点之后,需要将链表的头结点设置为null。
public class DeleteNode {
private static ListNode deleteNode(ListNode head, ListNode toBeDeleteNode) {
if (head == null || toBeDeleteNode == null) {
return head;
}
// 如果待删除的节点不是尾节点
if (toBeDeleteNode.getNext() != null) {
ListNode next = toBeDeleteNode.getNext();
toBeDeleteNode.setValue(next.getValue());
toBeDeleteNode.setNext(next.getNext());
} else if (head == toBeDeleteNode) { // 如果链表中只有一个节点,删除头结点
head = null;
} else { // 如果待删除的节点为尾节点,next指向null
ListNode index = head;
while (index.getNext() != toBeDeleteNode) {
index = index.getNext();
}
index.setNext(null);
}
return head;
}
public static void main(String[] args) {
ListNode head = Utils.createLinkedListWithoutOrder();
Utils.printList(head);
// 删除中间节点
ListNode midNode = SearchMidNode.searchMidNode(head);
System.out.println("MidNode is: " + midNode.toString());
deleteNode(head, midNode);
Utils.printList(head);
System.out.println("===========================");
// 删除只有一个节点的链表的头结点
ListNode head1 = new ListNode(3, null);
ListNode result = deleteNode(head1, head1);
Utils.printList(result);
System.out.println("===========================");
// 删除尾节点
ListNode tailNode = head;
while (tailNode.getNext() != null) {
tailNode = tailNode.getNext();
}
System.out.println("TailNode is: " + tailNode.toString());
deleteNode(head, tailNode);
Utils.printList(head);
}
}