牛客网高频算法题系列-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个结点后的新链表。

代码

  1. public class Bm009 {
  2.     /**
  3.      * 双指针法
  4.      *
  5.      * @param head ListNode类
  6.      * @param n    int整型
  7.      * @return ListNode类
  8.      */
  9.     public static ListNode removeNthFromEnd(ListNode head, int n) {
  10.         // 如果原链表为空，直接返回null
  11.         if (head == null) {
  12.             return null;
  13.         }
  14.         // 如果n不是正数，直接返回null
  15.         if (n <= 0) {
  16.             return null;
  17.         }
  18.         // 因为有可能删掉头结点，所以先设置一个假的头结点，并指向原有的头结点
  19.         ListNode dummyNode = new ListNode(-1);
  20.         dummyNode.next = head;
  22.         ListNode fast = dummyNode;
  23.         // 将fast结点指向第n-1个结点
  24.         for (int i = 0; i < n - 1 && fast != null; i++) {
  25.             fast = fast.next;
  26.         }
  27.         // 如果fast为null或者fast的next结点为空，说明链表长度小于n，不存在倒数第n个结点，直接返回null
  28.         if (fast == null || fast.next == null) {
  29.             return null;
  30.         }
  31.         ListNode slow = dummyNode;
  32.         // 快慢指针一起移动，直到fast移动到最后第二个结点，此时，slow即为倒数第n+1个结点
  33.         while (fast.next.next != null) {
  34.             slow = slow.next;
  35.             fast = fast.next;
  36.         }
  37.         // 将slow的next结点即倒数第n个结点移除
  38.         slow.next = slow.next.next;
  39.         return dummyNode.next;
  40.     }
  42.     public static void main(String[] args) {
  43.         ListNode pHead = ListNode.testCase2();
  44.         System.out.println("原链表为");
  45.         ListNode.print(pHead);
  46.         int k = 2;
  47.         System.out.println("删除链表中的倒数第" + k + "个结点后：");
  48.         ListNode.print(removeNthFromEnd(pHead, k));
  49.     }
  50. }

牛客网高频算法题系列-BM9-删除链表的倒数第n个节点 - 图1
牛客网高频算法题系列-BM9-删除链表的倒数第n个节点 - 图2
相信坚持的力量！