【算法训练】剑指offer#22 链表中倒数第K个节点

一、描述

链表中倒数第K个节点

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

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

示例 :

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

返回链表 4->5.

二、思路

• 先遍历整个链表，获取链表长度listLen，之后重新遍历，找到第listLen-k个节点

一次过...

三、解题

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None
class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        
        listLen = 0 # 链表默认长度为0
        tempNode = head
        # 如果链表为空，直接返回空链表
        if head is None:
            return head
        # 不为空
        else:
            listLen = 1 # 算上头节点长度至少为1
            while tempNode.next:
                listLen += 1
                tempNode = tempNode.next
            
        # 遍历后，再找到第listLen-k+1个节点即可
        for i in range(listLen-k):
            head = head.next
        
        return head