LeetCode25.给你一个链表,每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]
一、头插法
使用虚拟结点主要思路还是将其分成已经反转、正在反转和未反转三个部分。我们可以先遍历一遍数组,统计一下元素数量len,确定要分几组n=len/k,然后再进行链表反转。
public static ListNode reverseKGroup2(ListNode head, int k) {
ListNode dummyNode = new ListNode(0);
dummyNode.next = head;
ListNode cur = head;
int len = 0;//先计算出链表的长度
while(cur != null) {
len++;
cur = cur.next;
}
int n = len/k;//计算出有几组
ListNode pre = dummyNode;
cur = head;
for(int i=0;i<n;i++) {
for(int j=0;j<k-1;j++) {
ListNode next = cur.next;
cur.next = cur.next.next;
next.next = pre.next;
pre.next = next;
}
pre = cur;
cur = cur.next;
}
return dummyNode.next;
}
二、穿针引线法
新建虚拟头结点,然后我们直接遍历,根据是否为K个找到四个关键位置,并用变量pre、start、end和next标记,
接着就可以对上图中蓝颜色部分进行反转了,我们将end.net=nul,那蓝颜色部分可以直接进行反转链表。注意上图中指针的指向和几个变量的位置,head表示传给反转方法的参数。
之后我们要将裁下来的部分再缝到原始链表上,这就需要调整指针指向了,同样注意指针的指向。
最后调整一下标记,为下一次做准备
public static ListNode reverseKGroup(ListNode head, int k) {
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode pre = dummy;
ListNode end = dummy;
while (end.next != null) {
for (int i = 0; i < k && end != null; i++) {
end = end.next;
}
if (end == null)
break;
ListNode start = pre.next;
ListNode next = end.next;
end.next = null;
pre.next = reverse(start);
start.next = next;
pre = start;
end = pre;
}
return dummy.next;
}
private static ListNode reverse(ListNode head) {
ListNode pre = null;
ListNode curr = head;
while (curr != null) {
ListNode next = curr.next;
curr.next = pre;
pre = curr;
curr = next;
}
return pre;
}