描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例:
输入:{1,2,3,4,5},2
返回值:{4,5}
说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。
解题思路:
利用快慢指针思想,第一个指针先移动k步,然后第二个指针再从头开始,这个时候这两个指针同时移动,当第一个指针到链表的末尾的时候,返回第二个指针即可
代码:
struct ListNode* FindKthToTail(struct ListNode* pHead, int k ) {
// write code here
if (pHead == NULL)
return NULL;
struct ListNode* first = pHead;
struct ListNode* second = pHead;
//第一个指针先走k步
while (k-- > 0) {
if (first == NULL)
return NULL;
first = first->next;
}
//然后两个指针在同时前进
while (first != NULL) {
first = first->next;
second = second->next;
}
return second;
}