链表中倒数第k个节点
问题
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
解答
方法1(第一想法)
- 第一时间想到的解法:
先遍历统计链表长度,记为 n ;
设置一个指针走 (n-k) 步,即可找到链表倒数第 k 个节点。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
fin = sk = head
num = 0
while sk:
sk = sk.next
num = num + 1
num = num - k
for i in range(num):
fin = fin.next
return fin
方法2-使用双指针则可以不用统计链表长度
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