25、k个一组翻转链表

给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。

k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。

给定这个链表:1->2->3->4->5

当 k = 2 时,应当返回: 2->1->4->3->5

当 k = 3 时,应当返回: 3->2->1->4->5

方法一:递归法

思路:与(24、两两交换链表)中的思路类似,不过要注意链表的数量是否大于k

ListNode* dpReverse(ListNode* head, int k) {
	vector<ListNode*> vec;
	ListNode* p = head;
	for (int i = 0;i < k;++i) {
		if (p != NULL) {
			vec.push_back(p);
			p = p->next;
		}
		else {  //没有K个结点
			return head;
		}
	}
	ListNode* q = vec[vec.size() - 1];
	//翻转k个结点
	for (int i = vec.size() - 2;i >= 0;--i) {
		q->next = vec[i];
		q = q->next;
	}
	q->next = dpReverse(p, k);   //尾节点的next为后面链表的翻转
	return vec[vec.size() - 1];
}

ListNode* reverseKGroup(ListNode* head, int k) {
	if (head == NULL)
		return head;
	return dpReverse(head, k);
}

方法二:对于原地翻转链表的改进

在方法一中我采用vector对链表进行翻转,显然这会占用一定的内存空间。

//翻转k个结点(例如链表:1->2->3)head指向1,p指向3;翻转完成后变成3->2->1
ListNode* ptr = NULL, *tmp;
while (ptr != p) {
	tmp = head->next;
	head->next = ptr;
	ptr = head;
	head = tmp;
}

猜你喜欢

转载自blog.csdn.net/hello_noob/article/details/89838775