获取链表中倒数第K个节点

前言

给定一个单向链表的头节点，如何获取该链表中倒数第K个节点（从1开始计数）？本文将带着大家一起解决这个问题，欢迎各位感兴趣的开发者阅读本文。

在小程序中阅读

为了更好的阅读体验，你可以点击下方小程序来阅读本文。

思路分析

我们通过一个例子来做进一步的分析：

• 准备一个链表，它有6个节点，从头节点开始，其值依次为：1、3、5、9、15、21
• 获取该链表的倒数第3个节点

遍历两次链表

根据单向链表的定义，我们可知：想要获取它的某个节点，只能从头节点开始顺着其指针往后查找。假设整个链表有n个节点，那么倒数第K个节点就是从头节点开始的第n-K 1个节点。如果我们能够得到节点数n，那么只需要从头节点开始往后走n-k 1步就可以了。想要得到节点数n，就需要定义一个变量，从头开始遍历链表，每经过一个节点，这个变量就自增1。

也就是说，我们需要遍历链表两次，第一次计算出链表中节点的个数，第二次就能获取倒数第K个节点，如下图所示：

• 第1次遍历链表拿到了链表的长度n=6
• 第2次遍历链表获取到了倒数第3个节点处（6-3 1）的值9

IMG_498E0908B970-1

遍历一次链表

遍历两次链表来解决这个问题并不是最优解，我们换个思路来考虑下这个问题：准备两个指针。

• 第一个指针从链表的头部开始遍历向前走k-1（3-1=2）步，第二个指针保持不动
• 从第k步开始，第二个指针也开始从链表的头指针开始遍历，两指针同时向前走。
• 由于两个指针的距离始终保持在k-1，当第一个指针到达链表的尾节点时，第二个指针正好指向倒数第k个节点

IMG_596AE88489E9-1 2

实现代码

通过上面的分析，我们知道了如何用双指针的思路，只遍历一次链表就能获取链表的倒数第K个节点。接下来，我们来看下代码实现。

首先，我们设计一个名为GetLinkedListNode的类：

• 内部有2个私有变量
  • pNext P1指针
  • pHead P2指针
• 构造方法接受1个参数：链表头节点
  • 对参数进行校验
  • 修改两个指针的指向：默认指向链表头节点

代码语言：javascript复制

export class GetLinkedListNode {
  // p1指针
  private pNext: ListNode;
  // p2指针(与p1指针的距离始终保持在k-1)
  private pHead: ListNode;

  constructor(listHead: ListNode) {
    if (listHead == null) {
      throw new Error("链表头节点不能为空");
    }

    // 初始化两个指针指向
    this.pNext = listHead;
    this.pHead = listHead;
}

上述代码中，我们用了一个自定义类型ListNode，它描述了一个链表的节点应该包含哪些属性，对此感兴趣的开发者请移步我的另一篇文章：链表与变相链表的实现。

紧接着，实现获取倒数第K个节点函数：

• 接受一个参数K（从1开始），对参数进行有效性校验
• 修改p1指针的指向，将其指向k-1节点，k的范围也要做一下规避处理（其值大于链表总节点数）
• 同步修改p1、p2指针的指向，直至p1指针指向尾节点，p2指针正好指向倒数第K个节点

代码语言：javascript复制

  // 获取倒数第K个节点
  getCountdownNode(k: number): ListNode {
    if (k <= 0) {
      throw new Error("需要获取的倒数节点数必须大于0");
    }

    // p1指针先走，将其指向链表的k-1位置
    for (let i = 0; i < k - 1; i  ) {
      // k可能出现大于链表总节点的情况，需要进行规避
      if (this.pNext.next != null) {
        this.pNext = this.pNext.next;
      } else {
        throw new Error("需要找的节点不存在");
      }
    }

    // 两个指针同时向前走，直至p1指针指向尾节点
    while (this.pNext.next) {
      this.pNext = this.pNext.next;
      if (this.pHead.next) {
        this.pHead = this.pHead.next;
      }
    }

    // p2节点正好指向倒数第K个节点
    return this.pHead;
  }

完整代码请移步