题目:链表中倒数第k个结点
输入一个链表,输出该链表中倒数第k个结点。
思路1:一次遍历该链表,将结点一次存入列表res中,最后返回res[-k]。这里需要注意的是k的值是否符合题目要求,如果小于等于0或大于res长度则返回None。时间空间复杂度为O(n)。
代码:
# -*- coding:utf-8 -*-
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def FindKthToTail(self, head, k):
# write code here
if head is None:
return head
h = head
res = []
while h:
res.append(h)
h = h.next
if k > len(res) or k <=0:
return None
return res[-k]
思路2:设置两个指针指向头结点。然后快指针往前移动k-1步(这是为了两个指针之间相差k)。最后快慢指针同时前进,当快指针到达末尾时,慢指针正好指向倒数k结点上。
代码
if head is None or k <= 0:
return None
p = q = head
i = k - 1
while i > 0 and p:
p = p.next
i -= 1
if p is None:
return None
while p.next:
p = p.next
q = q.next
return q