5.1.10 10.在O(1)时间删除链表节点

优质
小牛编辑
130浏览
2023-12-01

一、题目

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

二、解题思路

由于给定了节点指针,那么要删除该节点。只要把该节点的值替换为下一个节点的值,同时让该节点直接指向下一个节点的下一个节点。相当于顶包代替了下一个节点,该节点自然就不存在。

需要注意的是如果指定节点是头结点,那么直接把头结点定义为下一个节点即可。如果是尾节点,需要循环遍历到该节点,然后让尾节点的上一个节点的指针为空即可。

三、解题代码

public class Test {  
    /** 
     * 链表结点 
     */  
    public static class ListNode {  
        int value; // 保存链表的值  
        ListNode next; // 下一个结点  
    }  

    /** 
     * 给定单向链表的头指针和一个结点指针,定义一个函数在0(1)时间删除该结点, 
     * 【注意1:这个方法和文本上的不一样,书上的没有返回值,这个因为JAVA引用传递的原因, 
     * 如果删除的结点是头结点,如果不采用返回值的方式,那么头结点永远删除不了】 
     * 【注意2:输入的待删除结点必须是待链表中的结点,否则会引起错误,这个条件由用户进行保证】 
     * 
     * @param head        链表表的头 
     * @param toBeDeleted 待删除的结点 
     * @return 删除后的头结点 
     */  
    public static 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.value = toBeDeleted.next.value;  
            // 待删除的结点的下一个指向原先待删除引号的下下个结点,即将待删除的下一个结点删除  
            toBeDeleted.next = toBeDeleted.next.next;  
        }  

        // 返回删除节点后的链表头结点  
        return head;  
    }  
}