JZ22 链表中倒数第k个节点

61 阅读1分钟

leetcode.cn/problems/li…

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

 

示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

解题思路:

第一时间想到的解法:

  1. 先遍历统计链表长度,记为 n ;
  2. 设置一个指针走 (n−k) 步,即可找到链表倒数第 k 个节点。

代码:

class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        int count = 0;
        ListNode temp = head;
        while (temp != null) {
            count++;
            temp = temp.next;
        }

        int n = count - k;
        for (int i = 0; i < n; i++) {
            head = head.next;
        }
        return head;
    }
}

使用双指针则可以不用统计链表长度。

image.png

算法流程:

  1. 初始化: 前指针 former 、后指针 latter ,双指针都指向头节点 head。
  2. 构建双指针距离: 前指针 former 先向前走 k 步(结束后,双指针 former 和 latter 间相距 k 步)。
  3. 双指针共同移动: 循环中,双指针 former 和 latter 每轮都向前走一步,直至 former 走过链表 尾节点 时跳出(跳出后, latter 与尾节点距离为 k−1,即 latter 指向倒数第 k 个节点)。
  4. 返回值: 返回 latter 即可。

复杂度分析:

时间复杂度 O(N) : N 为链表长度;总体看, former 走了 N 步, latter 走了 (N−k) 步。

空间复杂度 O(1) : 双指针 former , latter 使用常数大小的额外空间。

代码:

class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        ListNode former = head;
        ListNode latter = head;
        for (int i = 0; i < k; i++) {
            former = former.next;
        }
        while (former != null) {
            former = former.next;
            latter = latter.next;
        }
        return latter;
    }
}