【算法训练】剑指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