题目描述
输入一个链表,输出该链表中倒数第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