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

107 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第9天,点击查看活动详情

牛客网高频算法题系列-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));
    }
}

1.0136537.78343433291.01^{365} ≈ 37.7834343329
0.993650.025517964450.99^{365} ≈ 0.02551796445
相信坚持的力量!