思路:虽然是K个一组反转链表,但本质上还是反转链表。需要增加的就是一个循环,以及需要维护更多的节点位置。
在这里我加入了一个什么都不放的头节点。可以看到维护了四个指针,分别是beforePre(bp),beforeCur(bc),afterLast(al),afterNxt(an)。这四个分别表示什么呢?根据题意,第一次反转我们需要把 1 和 2 两个节点进行反转。所以在这里,我们把 节点1 和节点2 串起来的当做是一个子链表。bp表示子链表前面的一个节点,bc表示子链表第一个节点,al表示子链表最后一个节点,an表示子链表后面的一个节点。
所以下面的操作分为三个步骤:
- 将子链表进行反转;
- 将bp指向反转后的链表头;
- 将反转后的链表尾指向an。
可能大家会发现,上述操作貌似只用了bp,bc和an。那么al的作用在哪里呢?实际上如果我们想根据长度截取子链表,那么就需要an作为子链表结束的标志位了。
AC代码如下,有点丑:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
ListNode dummy = new ListNode();
dummy.next = head;
ListNode beforePre, beforeCur, afterLast, afterNxt, cur = dummy;
while(cur!=null) {
beforePre = cur;
beforeCur = beforePre.next;
for(int i=0; i<k; i++) {
cur = cur.next;
if(cur==null) {
break;
}
}
if(cur==null) break;
System.out.println(cur.val);
afterLast = cur;
afterNxt = afterLast.next;
ListNode pre = null;
ListNode beforeCurTmp = beforeCur;
while(beforeCur!=afterNxt) {
ListNode nxt = beforeCur.next;
beforeCur.next = pre;
pre = beforeCur;
beforeCur = nxt;
}
beforePre.next = pre;
// beforeCur.next = afterNxt;
cur = pre;
for(int i=0; i<k-1; i++) {
cur = cur.next;
}
cur.next = afterNxt;
}
return dummy.next;
}
}