題目描述
輸入一個鏈表,輸出該鏈表中倒數(shù)第k個結(jié)點。
思路
利用雙指針,一個指針先走k-1步,然后兩個指針同時開始走。第一個指針到尾部的時候,第二個指針指向倒數(shù)第k個結(jié)點。
代碼
class Solution:
def FindKthToTail(self, head, k):
if head == None or k <= 0:
return None
count_k = 0
p1 = head
p2 = head
while k > 1:
if p1.next != None:
p1 = p1.next
k = k - 1
else:
return None
while p1.next != None:
p1 = p1.next
p2 = p2.next
return p2