力扣 25.K个一组翻转链表
题目
给你链表的头节点 head
,每 k
个节点一组进行翻转,请你返回修改后的链表。
k
是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k
的整数倍,那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
- 示例1:
输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
- 示例2:
输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]
提示:
- 链表中的节点数目为
n
1 <= k <= n <= 5000
0 <= Node.val <= 1000
思路
采用多次局部翻转方法来完成,因此就需要单链表翻转函数来完成每次的翻转
具体步骤如下图所示:
这里偷懒一下,就没有画动图了,若有不清楚的地方可以评论区哦
如图上步骤所示
-
首先找到局部翻转的结尾位置,再将结尾结点的
idx -> next = NULL
,再从start
开始翻转 -
接着让
flag = idx
用于标记该位置,若是flag == NULL
则不进行操作,若是flag != NULL
,就要令flag->next = idx
(上图中就是倒数第三行时的情况,若是不令flag->next = idx
就会丢失4
,因为1的下一个是3而不是4
)
- 最后让
idx = start = temp
,简单来说就是初始化,初始化为下一次翻转的起始位置。
若是不清楚步骤,可以拿纸和笔手推一遍,这个方法我经常用到,能让思路变清晰,虽然很笨但很实用。
代码
//反转链表(三指针法)
//具体可以找到我的单链表博客,里面有翻转单链表的动图
//当时front和behind弄反了,但是不影响理解
//这里调整回来了
struct ListNode* reverseList(struct ListNode* head)
{
struct ListNode* front = head->next;
struct ListNode* behind = head;
while(front)
{
if(behind == head)
{
behind->next = NULL;
}
struct ListNode* temp = front->next;
front->next = behind;
behind = front;
front = temp;
}
head = behind;
return head;
}
struct ListNode* reverseKGroup(struct ListNode* head, int k)
{
//k==1时不需要翻转,直接返回该单链表即可
if(k == 1)
return head;
//初始化idx,start,flag
struct ListNode* idx = head;//用于遍历整个单链表
struct ListNode* start = head;//用于定位每次翻转的起始结点
struct ListNode* flag = NULL;//注!用于连接两次翻转之间的结点
while(1)
{
//找到翻转链表的结尾结点
for(int i=0;i<k-1;i++)
{
idx=idx->next;
//跳出条件
if(!idx)
return head;
}
//注意!这一个地方需要一点理解
if(flag)
flag->next = idx;
//temp用于定位下一次翻转的起始坐标
struct ListNode* temp = idx->next;
//开始翻转操作
idx->next =NULL;
reverseList(start);
//若是第一次翻转,需要把head重新定位一下
//比如2->1变成了1->2,此时head应该在1的位置
if(head == start)
head = idx;
//标记位置flag,需要理解,可以对照图片来推
flag = start;
//把链表先连接上
start->next = temp;
//初始化idx,start为下一次翻转的起始结点
start =temp;
idx = temp;
//跳出条件
if(!idx)
return head;
}
return head;
}
注意
1.flag需要认真理解,这一步是关键,否则会丢失结点
2.翻转单链表函数要是存在不理解问题,可以找到我的单链表博客,里面有动图解释
3.两个idx跳出循环条件都不可以丢,因为偷懒代码写出来还没有优化
题目来源: