剑指 Offer 22. 链表中倒数第k个节点

题目描述

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

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

示例：

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

返回链表 4->5.

---

本题很容易想到的一个做法就是遍历两边链表，第一遍记录链表长度len，然后再第二次遍历中数到第len-k个结点就返回。还有一种方法只需要遍历一边链表，那就是设置两个结点pre和cur，先只让cur开启遍历，当cur数到第k个结点时，pre也开始遍历，然后二者同时向后遍历，这样就保证了cur和pre中间隔了k个结点。最后当cur遍历到链表尾的时候，pre就指向了倒数第k个结点。代码如下：

class Solution(object):
    def getKthFromEnd(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        pre,cur=head,head
        count=0
        while cur:
            count+=1
            if count>=k:
                pre=pre.next
            cur=cur.next
        return pre