牛客网高频算法题系列-BM9-删除链表的倒数第n个节点

题目描述

给定一个链表，删除链表的倒数第 n 个节点并返回链表的头指针

原题目见：BM9 删除链表的倒数第n个节点

解法一：双指针法

首先，考虑两种特殊情况：

• 如果原链表为空，直接返回null。
• 如果k不是正数，直接返回null。

否则，使用双指针求解，求解过程如下：

• 因为有可能删掉头结点，所以先设置一个假的头结点dummyNode，并指向原有的头结点；
• 然后，遍历链表，将fast结点指向第n-1个结点；
• 如果遍历完后fast为null或者fast的next为空，说明链表长度小于n，不存在倒数第n个结点，直接返回null；
• 否则，快慢指针一起移动，直到fast移动到倒数第二个结点，此时，slow即为倒数第n+1个结点；
• 然后将slow的next结点即倒数第n个结点移除；
• 最后返回dummyNode.next即为删除第n个结点后的新链表。

代码

public class Bm009 {
    /**
     * 双指针法
     *
     * @param head ListNode类
     * @param n    int整型
     * @return ListNode类
     */
    public static ListNode removeNthFromEnd(ListNode head, int n) {
        // 如果原链表为空，直接返回null
        if (head == null) {
            return null;
        }
        // 如果n不是正数，直接返回null
        if (n <= 0) {
            return null;
        }
        // 因为有可能删掉头结点，所以先设置一个假的头结点，并指向原有的头结点
        ListNode dummyNode = new ListNode(-1);
        dummyNode.next = head;

        ListNode fast = dummyNode;
        // 将fast结点指向第n-1个结点
        for (int i = 0; i < n - 1 && fast != null; i++) {
            fast = fast.next;
        }
        // 如果fast为null或者fast的next结点为空，说明链表长度小于n，不存在倒数第n个结点，直接返回null
        if (fast == null || fast.next == null) {
            return null;
        }
        ListNode slow = dummyNode;
        // 快慢指针一起移动，直到fast移动到最后第二个结点，此时，slow即为倒数第n+1个结点
        while (fast.next.next != null) {
            slow = slow.next;
            fast = fast.next;
        }
        // 将slow的next结点即倒数第n个结点移除
        slow.next = slow.next.next;
        return dummyNode.next;
    }

    public static void main(String[] args) {
        ListNode pHead = ListNode.testCase2();
        System.out.println("原链表为");
        ListNode.print(pHead);
        int k = 2;
        System.out.println("删除链表中的倒数第" + k + "个结点后：");
        ListNode.print(removeNthFromEnd(pHead, k));
    }
}

相信坚持的力量！