算法学习-BM8:链表中倒数最后k个结点

算法学习

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;
	}
};