算法系列——在O(1)时间内删除链表结点(剑指offer)

题目描述

给定单向链表的头指针和一个结点指针，定义一个函数在O(1)时间内删除该结点。

解题思路

分为三种情况：
1.删除的是头结点

直接返回 head.next
2.删除的是尾结点

这种情况下，只能从头开始遍历，找到尾结点的前驱

3.删除的是中间结点

如果是中间节点，比如说 4->5->6->7->8 删除6对应的结点，
我们可以先将7赋给6，然后再将next指针指向8，那么就可以在O(1) 时间内删除这个结点。

核心思路代码如下所示：

 toBeDeleted.val= toBeDeleted.next.val;
  toBeDeleted.next = toBeDeleted.next.next;

程序实现

public class Solution {
    public  ListNode deleteNode(ListNode head, ListNode toBeDeleted) {
        // 如果输入参数有空值就返回表头结点  
        if (head == null || toBeDeleted == null) {
            return head;
        }
        // 如果删除的是头结点，直接返回头结点的下一个结点  
        if (head == toBeDeleted) {
            return head.next;
        }
        // 下面的情况链表至少有两个结点  
        // 在多个节点的情况下，如果删除的是最后一个元素  
        if (toBeDeleted.next == null) {
            // 找待删除元素的前驱  
            ListNode tmp = head;
            while (tmp.next != toBeDeleted) {
                tmp = tmp.next;
            }
            // 删除待结点  
            tmp.next = null;
        }
        // 在多个节点的情况下，如果删除的是某个中间结点  
        else {
            // 将下一个结点的值输入当前待删除的结点  
            toBeDeleted.val= toBeDeleted.next.val;
            // 待删除的结点的下一个指向原先待删除下下个结点，即将待删除的下一个结点删除  
            toBeDeleted.next = toBeDeleted.next.next;
        }
        // 返回删除节点后的链表头结点  
        return head;
    }

}