给出一个链表,每 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;
}