1、题目
输入一个链表,输出该链表中倒数第k个结点。链表的尾节点是倒数第一个节点。
2、思路
两个指针遍历链表:第一个指针从链表的头节点走k-1步,第二个指针不动;从第k步开始,第二个指针也开始从链表的头节点开始遍历,直到第一个指针指向链表结尾。由于两个指针的距离保持k-1,当第一个指针到达链表的尾节点时,第二个指针指向倒数第k个节点。
鲁棒性:
1.输入k=0时,输出nullptr
2.输入头指针为空时,输出nullptr
3.输入k大于链表长度时,输出nullptr
功能测试:
1.第k个节点在链表的中间
2.第k个节点是链表的头结点
3.第k个节点是链表的尾结点
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
// 查找倒数第0个节点和输入空指针
if(pListHead == NULL || k == 0){
return NULL;
}
// 两个指针遍历链表
ListNode *pAhead = pListHead;
ListNode *pBehind = pListHead;
// 第一个指针从链表的头结点走K-1步
for(unsigned int i = 0; i < k - 1; i++){
if(pAhead->next != NULL){
pAhead = pAhead->next;
}
else{
return NULL;
}
}
// 第k个节点开始,两个指针同时遍历
while(pAhead->next != NULL){
pAhead = pAhead->next;
pBehind = pBehind->next;
}
return pBehind;
}
};