在O(1)的时间复杂度下删除节点

题目

给定单向链表的头指针以及待删除的指针,定义一个函数在 O(1) 的时间复杂度下删除

解题思路

  1. 待删除节点非尾节点,将后一个节点的值复制到当前节点,然后删除后一个节点
  2. 待删除节点为尾节点,从头节点开始,找到待删除节点的前一个节点进行删除
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. }