在O(1)的时间复杂度下删除节点
题目
给定单向链表的头指针以及待删除的指针,定义一个函数在 O(1) 的时间复杂度下删除
解题思路
- 待删除节点非尾节点,将后一个节点的值复制到当前节点,然后删除后一个节点
- 待删除节点为尾节点,从头节点开始,找到待删除节点的前一个节点进行删除
1. public void O1DeleteNode(ListNode head, ListNode needDelete) {
3. if (needDelete.next != null) {
4. ListNode next = needDelete.next.next;
5. needDelete.val = needDelete.next.val;
6. needDelete.next = next;
8. } else {
9. ListNode cursor = head;
11. while (cursor != null) {
12. if (cursor.next == needDelete) break;
14. cursor = cursor.next;
15. }
17. if (cursor == null) return;
19. cursor.next = needDelete.next;
20. }
22. }
