链表中倒数第k个节点最简单的实现方式--python
程序员文章站
2022-08-09 15:49:21
思路:设置2个指针,第一个指针先走k步,第二个指针开始走,这样在第一个指针走到末尾时,第二个指针刚好指向倒数第K个节点class listnode(object): def __init__(self, x): self.val=x self.next=Nonedef findkth(head, k): if not head or not k: return p1,p2 = head, head for i in range(k-1): if p1...
思路:
设置2个指针,第一个指针先走k步,第二个指针开始走,这样在第一个指针走到末尾时,第二个指针刚好指向倒数第K个节点
class listnode(object):
def __init__(self, x):
self.val=x
self.next=None
def findkth(head, k):
if not head or not k:
return
p1,p2 = head, head
for i in range(k-1):
if p1.next == None: #这里要小心链表长度小于k的情况
return None
p1 = p1.next
while p1.next != None: #p1已经先走完,现在和P2开始一起走
p1 =p1.next
p2 =p2.next
return p2
测试:
node1 = listnode(1)
node2 = listnode(2)
node3 = listnode(3)
node4 = listnode(4)
node1.next = node2
node2.next = node3
node3.next = node4
new = findkth(node1, 2)
new.val
3
本文地址:https://blog.csdn.net/kasiko/article/details/107304617
下一篇: [整理]Python不同装饰器解释