剑指Offer 22. 链表中倒数第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即可得到节点
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
total = 0
cur = head
while cur:
cur = cur.next
total += 1
n = total - k
node = head
while n:
node = node.next
n -= 1
return node
- 双指针法。创建两个指针fast, slow,快指针先走k个,然后快指针和慢指针一起走,当快指针为空的时候,慢指针的位置即为结果。
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
fast, slow = head,head
while k:
fast = fast.next
k -= 1
while fast:
fast,slow = fast.next,slow.next
return slow