考点:链表
题目描述
输入一个链表,输出该链表中倒数第k个结点。
思路解析
快慢指针:快慢指针初始为头节点。快指针先走k-1步,到达第k个节点。此时慢指针相对快指针是倒数第k个节点。此时快慢指针一起走,当快指针到达最后一个节点时,慢指针位于倒数第k个节点。
如K为3,则pA先走2步,先到第3个节点。pA、pB一起走。当pA到最后一个节点时,pB到倒数第3个节点
c++实现
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if(pListHead==NULL || k<=0)
return NULL;
ListNode* pA=pListHead;
ListNode* pB=pListHead;
for(int i=0;i<k-1;i++){//走k-1步
if(pA->next !=NULL){ //pA下一个不为空,继续走;若为空,说明不存在倒数第k个
pA=pA->next;
}
else
return NULL;
}
while(pA->next !=NULL){ //走到最后一个节点停下
pA=pA->next;
pB=pB->next;
}
return pB;
}
python实现
class Solution:
def FindKthToTail(self, head, k):
# write code here
if(head==None or k<=0):
return None
pA=head
pB=head
for i in range(k-1):
if pA.next != None:
pA=pA.next
else:
return None
while(pA.next !=None):
pA=pA.next
pB=pB.next
return pB
方法2:python切片
class Solution:
def FindKthToTail(self, head, k):
# write code here
res=[]
while head != None: #最后一个结点不为空,也要存起来
res.append(head)
head=head.next
if len(res)<k or k<=0:
return None
return res[-k] #切片