题目描述
输入一个链表,输出该链表中倒数第k个结点。
分析
快慢指针法解决,设置两个指向头节点的指针,让其中一个先走k步,然后两个指针同时向后遍历,直到快指针为空时循环终止,此时慢指针指向的就是倒数第k个节点。
例:1,2,3,4,5,6,7;设 k = 2;初始化 i = 1;j = 1; j向前至 2此时指向3,然后i,j同时向后,直至j指向null,此时i指向6,即倒数第二个位置。
本题需要判断输入的k是否合法,非法输入即k的值大于链表长度。
C++代码如下
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
ListNode*p1 = pListHead;
ListNode*p2 = pListHead;
if(p1 == NULL) return NULL;
int n = k;
while(p1&&n){ //判断输入k是否小于链表长度;
n--;
p1 = p1->next;
}
if(p1 == NULL && n>0) return NULL;
p1 = pListHead;
while(p1&&k>0){ //p1先向前走k步
k--;
p1 = p1->next;
}
while(p2&&p1){ //p1和p2同时后移,直到有一个为空
p1 = p1->next;
p2 = p2->next;
}
return p2;
}