leetcode算法学习(16)—— 链表中倒数第k个节点

题目描述

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。例如，一个链表有6个节点，从头节点开始，它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

示例：
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.

方法一：计算链表长度

思路

• 最直接的想法：
  • 遍历统计链表长度，记为n；
  • 设置一个指针走 (n-k)步，即可找到链表倒数第 k个节点。
    

代码

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        n,i=0,0
        temp=head
        while head:
            head=head.next
            n+=1

        while i<n-k:
            temp=temp.next
            i+=1
        return temp

方法二：双指针

思路

• 初始化： 前指针 former 、后指针 latter ，双指针都指向头节点 head​
• 构建双指针距离： 前指针 former 先向前走k步（结束后，双指针 former 和 latter 间相距 k步。
• 双指针共同移动： 循环中，双指针 former 和 latter 每轮都向前走一步，直至 former走完节点，latter相当于走了n-k步。
• 返回值： 返回 latter 即可。

代码

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        former,latter=head,head
        for _ in range(k):
            former=former.next
        
        while former:
            former,latter=former.next,latter.next
        return latter