给你一个链表,每 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.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
ListNode pre = head;
int count = 0;
while(pre!=null)
{
count++;
pre = pre.next;
}
count = count/k;
pre = null; //上一个k链的尾部,初始为空
ListNode dummy = null; //虚拟的头节点
ListNode newHead = null; //保存返回值
while(count!=0)
{
ListNode tail = null;
//翻转后,头部会变成尾部节点,因此先保存头节点用于更新尾部
tail = head;
//翻转k个节点,采用头部插入的方法
for(int i=0;i<k;i++)
{
ListNode node = head;
head = head.next;
node.next = dummy;
dummy = node;
}
//将翻转后的k长链表,与上一个链表的尾部连接
//如果没有上一个,需要设置头部
if(pre==null) newHead = dummy;
//有上一个,则上一个的下一个就是当前链表的头节点
else pre.next = dummy;
//清除虚拟头节点
dummy = null;
//更新尾部
pre = tail;
count--;
}
pre.next = head;
return newHead;
}
}