题意
给定一个链表的头指针和一个整数 k,找到链表的倒数第 k 个节点,如果 k 大于链表长度,则返回 NULL
分析
暴力做法是先遍历一次统计长度,然后算出倒数第 k 个的正序,然后再遍历一次。
优化做法:先让一个指针前进到第 k 个节点,然后再用一个新的指针指向头节点,之后两个指针一起向后走,前面的指针到最后一个节点的时候第二个指针就到了倒数第 k 个节点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* findKthToTail(ListNode* pListHead, int k) {
if(pListHead == NULL)return pListHead;
ListNode* cur = pListHead;
while(cur->next != NULL && k > 1){
cur = cur->next;
k --;
}
if(k > 1)return NULL;
ListNode* res = pListHead;
while(cur->next != NULL){
cur = cur->next;
res = res->next;
}
return res;
}
};