[剑指Offer] 22_链表中导数第K个节点

题目

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

例：

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

---

思路

1. 完成功能很简单，双指针遍历链表，快指针比慢指针快K个节点，当快指针为None时，慢指针即为倒数第K个节点。
   1. 时间复杂度：O(n)
   2. 空间复杂度：O(1)
2. 在此给出一个比较单纯的解法，遍历一次链表记录长度，计算出是正数第几个，再度遍历。
   1. 时间复杂度：O(n)，虽然都是O(n)，此解法需要遍历2次，系数显然是大于思路1的
   2. 空间复杂度：O(1)

---

代码

思路1：时间复杂度：O(n)，空间复杂度：O(1)

def kth_node_from_end(head, k):
    """    
    :param head: head of listnode
    :param k: kth from end
    :return: node
    """
    if head:
        slow = fast = head
    else:
        raise Exception('EMPTY INPUT')

    for i in range(k):
        if fast:
            fast = fast.next
        else:
            raise Exception("K Out Of Range")

    while fast:
        fast = fast.next
        slow = slow.next

    return slow

---

思考

正如书上所说，这道题考察重点还是在防范异常输入上。

1. head is None，如果此时K>0，会出现None.next报错
2. K大于链表长度，在fast node 先走的循环中就会出现None.next
3. 关于循环次数的，由于Python数字时没有长度限制的，因此不会出现

附上LeetCode题目，供检验功能性。

LeetCode 19. 删除链表的倒数第N个节点

题目

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        fast = slow = head
        for i in range(n):
            fast = fast.next        
        if not fast:
            return slow.next
        while fast.next:
            slow = slow.next
            fast = fast.next
        slow.next = slow.next.next
        return head