LeetCode 剑指 Offer——链表中倒数第K个节点、反转链表

链表中倒数第K个节点

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

示例：

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

返回链表 4->5.

代码：

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

时间复杂度O(n)，空间复杂度O(1)。用两个相距K的指针，一前一后同步向后遍历，当前指针到达尾部后指针即为输出。

反转链表

定义一个函数，输入一个链表的头节点，反转该链表并输出反转后链表的头节点。

示例：

输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL

代码：

class Solution:
    def reverseList(self, head: ListNode) -> ListNode:
        if head == None or head.next == None:
            return head
        p1, p2 = head, head.next
        p1.next = None  
        while p2.next != None:
            tmp = p2 
            p2 = p2.next 
            tmp.next = p1
            p1 = tmp 
        p2.next = p1 
        return p2

时间复杂度O(n)，空间复杂度O(1)。用两个指针一前一后同步向后遍历，同时修改连接关系即可。