日期:2021年4月1日标签:Algorithm/Math

求链表中倒数第K个结点 #

一.单向链表 #

维基百科中国是这样描述的:

单向链表(又名单链表、线性链表)是链表的一种,其特点是链表的链接方向是单向的,对链表的访问要通过从头部开始,依序往下读取。

从描述中,我们能知道单向链表的方向是单一的,也就是我们只能从第一个结点索引到第二个结点,无法从第二个结点索引到第一个结点。单链表在内存中是一组地址任意的存储单元,与数组不同,单链表是通过指针索引,每个结点除了存储当前结点的值,还会存储下一个结点的指针,尾结点的指针为空。 单链表

二.分析 #

因为单链表访问顺序是单向的,所以如果是求第K个结点,我们只要从头开始每次指针向右移动一个位置,移动K-1次,就能找到第K个结点。但是题目是倒数第K个。

留给大家一点思考的时间......

...............................................

..............................................

相信大家都已经知道方法了,没错很简单,就是利用双指针的思路。

  1. 两个指针开始都位于0位置。
  2. 将第二个指针移动到K-1的位置。
  3. 然后两个指针同时移动,直到最后一个指针移动到链表的尾结点,此时第一个指针正好指在第K个结点。

方法很简单,但是要提醒的是要注意判断一些临界条件,考虑K是否超出链表的长度。否则写出来的代码没有鲁棒性。

三.解题 #

通过上面的分析,我们很容易写出下面的代码。代码采用typescript写的。


/**
 * 找到单链表中倒数第k个结点。
 * 本例子采用的是数字链表,实际上可以是存储任意类型的值。
 */
// 链表的结点类
class ListNode {
    public value: number | null;
    public next: ListNode | null;
    constructor(value, next) {
        this.value = value;
        this.next = next;
    }
}
// 创建一个链表
const createLinkList = (values: number[]) => {
    const head = new ListNode(null, null);
    let preNode = head;
    for (let i = 0; i < values.length; i++) {
        const currentNode = new ListNode(values[i], null);
        preNode.next = currentNode;
        preNode = currentNode;
    }
    return head;
};
// 获得倒数第k个值
const getKthFromLast = (k: number, linkHead: ListNode) => {
    if (k < 1) throw new Error("k值必须大于0");
    // 开始时p1和p2都指向head结点
    let p1 = linkHead, p2 = linkHead;
    // 先将p2移动到k-1的位置
    for (let i = 0; i < k - 1; i++) {
        if (p2.next !== null) {
            p2 = p2.next;     
        } else {
            return null;
        }
    }
    // 同时移动k1和k2,直到p2移动到尾结点
    while (p2.next !== null) {
        p1 = p1.next;
        p2 = p2.next;
    }
    return p1.value;
};
// 测试
const linkList1 = createLinkList([1, 2, 3, 4, 5, 6]);
const linkList2 = createLinkList([-1, -3, -10, 10, -1, 3]);
const linkList3 = createLinkList([]);
console.log(getKthFromLast(1, linkList1));
console.log(getKthFromLast(2, linkList1));
console.log(getKthFromLast(6, linkList1));
console.log(getKthFromLast(7, linkList1));
console.log(getKthFromLast(1, linkList2));
console.log(getKthFromLast(6, linkList2));
console.log(getKthFromLast(8, linkList2));
console.log(getKthFromLast(1, linkList3));
console.log(getKthFromLast(-1, linkList3));

测试结果如下:

如有不正确或者遗漏的地方欢迎指出。

github地址

目录