K 个一组翻转链表
题目描述
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
思路
- 本题相比较于链表的翻转,多了一个分段的过程。要求K个一段,最后不足K个的不分段。
- 为了确定该段是不是需要翻转(即该段的个数是不是小于K),以及该段的终点。因此在翻转链表之前需要一个函数获取每一段的终点,但返回nullptr时表示个数不满足要求,也就意味着此段不需要翻转。
- 在翻转的过程中需要一个段与段之间的连接也是一个必须要考虑的问题。这里要求每次翻转完成之后,返回翻转之后链表的最后一个结点。显而易见,翻转之后的最后一个结点,就是翻转之前的第一个结点。
实现
首先是寻找断尾,也就是确定翻转的界限。
传递进去的参数head就是该段的段头的指针,k就是题目中k的含义,也就是说是每一段的最大长度。
最后返回的值是该段最后一个元素的指针,若该段的个数不足k个那么将返回NULL。
ListNode* getSingleRear(ListNode* head,int k){
int length = 1;
ListNode* p = head;
while(p && length < k){
p = p->next;
length++;
}
return p;
}
翻转链表的函数:
- head,使用头插法实现,head就相当于是头结点,也就是上一段的最后一个元素的指针。
- start, 翻转的起始位置的指针。
- end, 翻转的终点位置的后一个结点。
- 返回值为翻转之后,该段的最后一个结点的指针。
- 若翻转的长度为1(表现为start->next == end),不需要翻转,返回即可。
ListNode* reverse(ListNode* head,ListNode* start,ListNode* end){
if(start->next == end){
head->next = start;
}
else{
ListNode* p = start;
while(p != end){
ListNode* q = p;
p = p->next;
q->next = head->next;
head->next = q;
}
}
return start;
}
主函数中维护三个变量
- ph,记录当前正在执行翻转的段的头结点,初始时其指向一个新建的头结点。
- p, 记录当前正在执行段的第一个元素的指针。
- nex, 记录下一个将要执行的段的第一个元素的指针。这个指针也意味着是当前正在执行段的最后一个结点的尾结点的后一个结点。
值得说明的是,因为要明确返回值为NULL时的含义,所以在getSingleRear中返回的就是当前正在执行段的尾结点,与实际nex的意义不相符合。因此需要执行nex = nex->next。
在不需要翻转时,直接将剩余的链表插入到ph之后即可。
ListNode* reverseKGroup(ListNode* head, int k) {
if(!head){
return head;
}
ListNode newHead(0);
ListNode* p = head;
ListNode* ph = &newHead;
ListNode* nex;
do{
nex = getSingleRear(p,k);
if(nex){
nex = nex->next;
ph = reverse(ph, p, nex);
p = nex;
}
else{
ph->next = p;
}
}while(nex);
return newHead.next;
}