题目描述
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
解题思路
/**
* 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) {
ListNode *L = head;
ListNode *res = new ListNode(0);
ListNode *pre = res;
res->next = head;
while(L)
{
int n = k;
ListNode *inside = L;
// 检查是否可以翻转
while(n && inside)
{
inside = inside->next;
n--;
}
if(n == 0)
{
//说明符合k哪么反转就OK了
n = k;
ListNode *p1 = inside;
ListNode *p2 = L;
ListNode *p3 = p2;
while(n)
{
p3 = p2->next;
p2->next = p1;
p1 = p2;
p2 = p3;
n --;
}
pre->next = p1;
pre = L;
}
L = inside;
}
return res->next;
}
};