本文讲解返回倒数第k个节点的两种方法
目录
题目链接:返回倒数第 k 个节点
法一、遍历链表法
解题思路:
求出链表的长度n,找到倒数第k个节点,只需从第一个节点往后走n-k步即可找到这个节点。
class Solution {
public:
int kthToLast(ListNode* head, int k) {
ListNode* cur = head;
int n = 0;
//求出链表长度
while (cur)
{
cur = cur->next;
n++;
}
cur = head;//重置
for (int i = 0; i < n - k; i++)
cur = cur->next;
return cur->val;
}
};
法二、快慢指针法
解题思路:
这种方法只循环遍历了一次链表,效率更高
定义两个快慢指针:fast(快指针),slow(慢指针)。求倒数第k个结点,初始条件使fast和slow均指向第一个节点。对于slow,只需要正走n-k步(n为总长度)就是倒数第k个结点(可以画图悉知),此方法是在不知道总长度n的情况下进行的(因为这种方法要求只能遍历链表一次),所以我们才定义了一个快指针fast,让其先走k步,那么走完之后,此时fast再走到NULL时,正好为n-k步。
总体思路:slow和fast初始均指向第一个节点,再让fast先走k步,然后slow和fast再同时走,直到fast为NULL,此时slow所指向的节点就是倒数第k个结点。(因为此时正好满足slow正走n-k步)
class Solution {
public:
int kthToLast(ListNode* head, int k) {
ListNode* fast = head, *slow = head;
while (k--)
{
//假设题目中没告诉k一定有效,那么就要加一句下面的代码
//if (fast == NULL) return -1;//返回什么看题目要求,一般为-1
//如果k>链表长度,会提前变为NULL,那么直接返回NULL,这种情况对空链表也适用
fast = fast->next;
}
while (fast)
{
fast = fast->next;
slow = slow->next;
}
return slow->val;
}
};