【数据结构与算法】查找带头结点的单链表中倒数第k个结点的算法 C++实现(单链表+双指针法)

avatar
作者
筋斗云
阅读量:0

设L是一个带头结点的单链表的表头指针,结点结构为(data,Link),在不改变链表的前提下,设计一个算法,查找链表中倒数第k个结点(k为正整数)。若查找成功,算法输出该结点的元素值,并返回1;否则,只返回0。


思路

双指针法,同时使用两个指针p1和p2。两个指针都指向链表的头节点。然后,指针p1开始遍历链表,每次前进一步。当p1前进到第k+1个节点时,指针p2开始从头节点开始遍历。这样,p1和p2始终保持k个节点的距离。当p1到达链表的末尾时,p2刚好位于链表的倒数第k个节点。

声明两个指针p1和p2,并将它们都初始化为指向链表的头节点。然后,使用一个for循环遍历链表。在每次迭代中,p1都向前移动一步。当p1移动了k步后,p2开始移动。

如果链表的长度小于k,那么p1将在p2开始移动之前到达链表的末尾。在这种情况下,函数返回ERROR,表示无法找到倒数第k个元素。

如果链表的长度大于或等于k,那么当p1到达链表的末尾时,p2将指向链表的倒数第k个元素。然后,函数打印p2指向的节点的数据,并返回OK,表示成功找到了倒数第k个元素。

由于每个节点只被访问一次,因此时间复杂度为 O ( n ) O(n) O(n),其中 n n n是链表的长度。这个算法只使用了常量级别的额外空间(两个指针p1和p2),所以空间复杂度为 O ( 1 ) O(1) O(1)


代码

#include <algorithm> #include <iostream> #define AUTHOR "HEX9CF" using namespace std; using Status = int; using ElemType = int;  const int N = 1e6 + 7; const int TRUE = 1; const int FALSE = 0; const int OK = 1; const int ERROR = 0; const int INFEASIBLE = -1; const int OVERFLOW = -2;  int n; ElemType a[N];  struct ListNode { 	ElemType data; 	ListNode *next; }; using LinkedList = ListNode *;  Status initList(LinkedList &first) { 	first = (ListNode *)malloc(sizeof(ListNode)); 	first->next = NULL; 	return OK; }  Status listInsert(LinkedList &L, int pos, ElemType e) { 	ListNode *p = L; 	for (int i = 0; p && i < pos; i++) { 		p = p->next; 	} 	if (!p) { 		return ERROR; 	} 	ListNode *newNode = (ListNode *)malloc(sizeof(ListNode)); 	newNode->data = e; 	newNode->next = p->next; 	p->next = newNode; 	return OK; }  ElemType getElem(LinkedList L, int pos) { 	ListNode *p = L->next; 	for (int i = 0; p && i < pos; i++) { 		p = p->next; 	} 	return p->data; }  ElemType getNthLastElem(LinkedList L, int k) { 	int i; 	ListNode *p1, *p2; 	p1 = p2 = L->next; 	for (i = 0; p1; i++) { 		p1 = p1->next; 		if (i > k) { 			p2 = p2->next; 		} 	} 	if (i < k) { 		return ERROR; 	} 	cout << p2->data << "\n"; 	return OK; }  int main() { 	cin >> n; 	for (int i = 0; i < n; i++) { 		cin >> a[i]; 	}  	int k; 	cin >> k;  	LinkedList L; 	initList(L);  	for (int i = 0; i < n; i++) { 		listInsert(L, i, a[i]); 	}  	// for (int i = 0; i < n; i++) { 	// 	cout << getElem(L, i) << " "; 	// } 	// cout << "\n";  	cout << getNthLastElem(L, k) << "\n"; 	return 0; }  

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!