给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
如果之前没有解过两两为一组翻转链表的题,请先参考我的博客之 两两交换链表中的节点
算法思路: k个交换顺序(前后倒置),应该不难想到利用栈的“先进后出”的特性,进行倒置。再结合两两交换链表中的节点中的递归算法思路,那么这道题的思路就清晰了。
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
ListNode* reverseKGroup(ListNode* head, int k) {
if (k <= 1){//首先判断k的合法性
return head;
}
stack<ListNode *> myStack;//辅助栈,用于转换k个节点的转换
int count = 0;//进栈的节点数
ListNode *ptr = head;
//进栈k个节点
while (count < k && ptr != NULL){
myStack.push(ptr);
ptr = ptr->next;
++count;
}
if (count < k){//如果不足k个,则无需转换
return head;
}
//出栈并且顺序串接好
//pHead是逆置结果的表头,pEnd是逆置结果的临时表尾
ListNode *pHead = myStack.top(), *pEnd;
myStack.pop();
pEnd = pHead;
while (count > 1){
pEnd->next = myStack.top();
pEnd = pEnd->next;
myStack.pop();
--count;
}
//递归调用,将ptr后面的链表进行逆置
pEnd->next = reverseKGroup(ptr, k);
return pHead;
}
至于非递归的算法,代码不是很简洁,此处不再累赘。