把链表节点按照k个一组进行分组,用一个指针head指向每组的头节点,head每次向前移动k步,直到链表尾部。
对于每一个组,先判断这个组的节点的个数是否大于等于k,如果是的再翻转,如果不是的话,就不需要翻转了。
每一个链表反转后,头部还需要与上一个子链表连接,尾部需要与下一个链表连接。
思路:
定义一个新的节点pre,让pre的next指向head。然后反复移动head和pre,对head指向的子链表进行翻转,直到结尾。
翻转链表:(并让链表的尾部指向下一个子链表的头部)
public ListNode[] myReverse(ListNode head, ListNode tail) {
ListNode prev = tail.next;
ListNode p = head;
while (prev != tail) {
ListNode nex = p.next;
p.next = prev;
prev = p;
p = nex;
}
return new ListNode[]{tail, head};
}
执行每一组的翻转:
public ListNode reverseKGroup(ListNode head, int k) {
ListNode hair = new ListNode(0);
hair.next = head;
ListNode pre = hair;
while (head != null) {
ListNode tail = pre;
// 查看剩余部分长度是否大于等于 k
for (int i = 0; i < k; ++i) {
tail = tail.next;
if (tail == null) {
return hair.next;
}
}
ListNode nex = tail.next;
ListNode[] reverse = myReverse(head, tail);
head = reverse[0];
tail = reverse[1];
// 把子链表重新接回原链表
pre.next = head;
tail.next = nex;
pre = tail;
head = tail.next;
}
return hair.next;
}