深圳幻海软件技术有限公司 欢迎您!

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

2023-02-28

前言给定一个单向链表的头节点,如何获取该链表中倒数第K个节点(从1开始计数)?本文将带着大家一起解决这个问题,欢迎各位感兴趣的开发者阅读本文。思路分析我们通过一个例子来做进一步的分析:准备一个链表,它有6个节点,从头节点开始,其值依次为:1、3、5、9、15、21。获取该链表的倒数第3个节点。遍历两

前言

给定一个单向链表的头节点,如何获取该链表中倒数第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。

遍历一次链表

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

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

实现代码

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

首先,我们设计一个名为GetLinkedListNode的类:

  • 内部有2个私有变量。

pNext P1指针;

pHead P2指针。

  • 构造方法接受1个参数:链表头节点。

对参数进行校验;

修改两个指针的指向:默认指向链表头节点。

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;
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.

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

紧接着,实现获取倒数第K个节点函数:

  • 接受一个参数K(从1开始),对参数进行有效性校验。
  • 修改p1指针的指向,将其指向k-1节点,k的范围也要做一下规避处理(其值大于链表总节点数)。
  • 同步修改p1、p2指针的指向,直至p1指针指向尾节点,p2指针正好指向倒数第K个节点。
// 获取倒数第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;
  }
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.
  • 25.
  • 26.
  • 27.

完整代码请移步:GetLinkedListNode.ts。

小tips:我们在写代码的时候,一定要尽可能地考虑各种边界情况的处理,最大程度的避免一些错误的发生,提升代码的健壮性。

例如上述代码中所做的处理,最大程度的避免了程序因取不到值而引发的异常报错问题,我们也管这种做法称为防御性编程。

这是一种良好的编程习惯,在编写代码的时候多问问自己“如果不······那么······”这样的问题,提前预见在什么地方可能会出现问题,并为这些可能出现的问题制定处理方式。这样,当异常情况发生时,软件的行为也尽在我们的掌握之中,而不至于出现不可预见的事情。

测试用例

接下来,我们将前言中的例子代入上个章节所实现的函数中,验证下它能否得出正确的结果。

const linkedList = new LinkedList();
linkedList.push(1);
linkedList.push(3);
linkedList.push(5);
linkedList.push(9);
linkedList.push(15);
linkedList.push(21);

const getLinkedListNode = new GetLinkedListNode(linkedList.getHead());
const resultNode = getLinkedListNode.getCountdownNode(3);
console.log(resultNode);
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.

运行结果如下所示,成功的解决了文章前言中所讲的问题。

完整代码请移步:GetLinkedListNode-test.ts。

注意:上述代码中用到的LinkedList是自定义的一个类,它实现了链表这个数据结构。

示例代码

本文所列举的代码,其完整版请移步:

  • GetLinkedListNode.ts
  • GetLinkedListNode-test.ts