算法学习
BM8 链表中倒数最后k个结点
https://www.nowcoder.com/practice/886370fe658f41b498d40fb34ae76ff9?tpId=295&sfm=html&channel=nowcoder
题目描述:输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。如果该链表长度小于k,请返回一个长度为 0 的链表。
思路:
快慢双指针
1.准备一个快指针,从链表头开始,在链表上先走步。
2.准备慢指针指向原始链表头,代表当前元素,则慢指针与快指针之间的距离一直都是。
3.快慢指针同步移动,当快指针到达链表尾部的时候,慢指针正好到了倒数个元素的位置。
代码实现:
#include <iostream>
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
ListNode* FindKthToTail(ListNode* pHead, int k) {
ListNode* fast = pHead;//设置快慢指针(初始头指针)
ListNode* slow = pHead;
for (int i = 0; i < k; i++) //快指针先走k步
{
if (fast != NULL)
fast = fast->next;
else
return slow = NULL; //k步内有NULL,说明链表长度小于k,返回NULL
}
//快慢同时移动,快指针先到末尾,此时慢指针指向倒数第k个
while (fast != NULL)
{
fast = fast->next;
slow = slow->next;
}
return slow;
}
};