给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k)
{
if (head == nullptr)
return nullptr;
// 区间 [a, b) 包含 k 个待反转元素
ListNode* a;
ListNode* b;
a = b = head;
for (int i = 0; i < k; i++)
{
// 不足 k 个,不需要反转,base case
if (b == nullptr)
return head;
b = b->next;
}
// 反转前 k 个元素
ListNode* newHead = reverse(a, b);
cout << newHead->val << endl;
// 递归反转后续链表并连接起来
a->next = reverseKGroup(b, k);
return newHead;
}
ListNode* reverse(ListNode* a, ListNode* b)
{
ListNode* pre;
ListNode* cur;
ListNode* nxt;
pre = nullptr;
cur = a;
nxt = a;
// while 终止的条件改一下就行了
while (cur != b)
{
nxt = cur->next;
cur->next = pre;
pre = cur;
cur = nxt;
}
// 返回反转后的头结点
return pre;
}
};