输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
第一种做法,简单,就是先遍历链表长度,然后输出正向的第n-k+1个。
class Solution(object):
def getKthFromEnd(self, head, k):
"""
:type head: ListNode
:type k: int
:rtype: ListNode
"""
if not head:
return
a = 0
phead = head
phead1 = head
while phead != None:
a += 1
phead = phead.next
for i in range(1,a-k+1):
phead1 = phead1.next
return phead1
第二种做法是,通过双指针,也就是两个指针相间隔k,如果后面的指针先到了最后,前面的指针也就到达了倒数第K个。
想法虽简单,但是可能会弄错位置!
结束条件是 phead1.next != None。
如果是 phead1 != None ,则后面的指针已经超过了原来的链表,不是最后一个结点。
class Solution(object):
def getKthFromEnd(self, head, k):
if not head:
return
phead1, phead2 = head, head
a = 1
while a < k:
phead1 =phead1.next
if phead1 == None:
return
a += 1
while phead1.next != None:
phead2 = phead2.next
phead1 = phead1.next
return phead2