剑指 Offer 18_1 删除链表节点

1 题目

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

2 提示

• 题目保证链表中节点的值互不相同；
• 若使用 C 或 C++ 语言，你不需要 free 或 delete 被删除的节点。

3 示例

// 示例 1
输入: head = [4,5,1,9]，delete=4
输出: [5,1,9]

// 示例 2
输入: head = [4,5,1,9]，delete=5
输出: [4,1,9]

// 示例 3
输入: head = [4,5,1,9]，delete=9
输出: [4,5,1]

// 示例 4
输入: head = [9], delete=9
输出: null

4 解题思路

• 对于非尾节点，把下一个节点的内容覆盖要删除的节点，并删除下一个节点；
• 对于尾节点，先顺序查找，再删除。

5 复杂度分析

• 时间复杂度：O(n)
• 空间复杂度：O(1)

6 题解

/**
 * public class ListNode {
 *   int val;
 *   ListNode next;
 *   ListNode() {}
 *   ListNode(int val) { this.val = val; }
 *   ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
public class JZ18_1_删除链表节点 {

    public static void main(String[] args) {
        // 测试用例
        for (int i = 0; i < 4; i++) {
            ListNode<Integer> node9 = new ListNode(9);
            ListNode<Integer> node1 = new ListNode(1, node9);
            ListNode<Integer> node5 = new ListNode(5, node1);
            ListNode<Integer> node4 = new ListNode(4, node5);
            if (i == 0) {
                // 删除头结点
                PrintUtils.getInstance().printListNode(node4);
                PrintUtils.getInstance().printListNode(solution(node4, node4), "打印删除节点4后的链表");
            } else if (i == 1) {
                // 删除中间节点
                PrintUtils.getInstance().printListNode(node4);
                PrintUtils.getInstance().printListNode(solution(node4, node5), "打印删除节点5后的链表");
            } else if (i == 2) {
                // 删除尾节点
                PrintUtils.getInstance().printListNode(node4);
                PrintUtils.getInstance().printListNode(solution(node4, node9), "打印删除节点9后的链表");
            } else if (i == 3) {
                // 删除节点，有且仅有一个节点
                PrintUtils.getInstance().printListNode(node9);
                PrintUtils.getInstance().printListNode(solution(node9, node9), "打印删除节点9后的链表");
            }
        }
    }
    
    /**
     * 时间复杂度：O(n)
     * 空间复杂度：O(1)
     */
    private static ListNode solution(ListNode head, ListNode delete) {
        if (null == head) return head;
        if (head == delete) return head.next;

        ListNode pre = head, curr = head.next;
        while (null != curr && curr != delete) {
            pre = curr;
            curr = curr.next;
        }
        if (null != curr) pre.next = curr.next;
        return head;
    }
}