K个一组反转链表解题思路:
首先可以抽取一个函数,这个函数用于将需要反转的一段区域进行反转:
public ListNode[] reverse(ListNode head,ListNode tail){
ListNode pre=tail.next,cur=head,next=null;
while(pre!=tail){
next=cur.next;
cur.next=pre;
pre=cur;
cur=next;
}
return new ListNode[]{tail,head};
}
这里的head就是需要反转的区域的第一个节点,tail是需要反转区域的最后一个节点
定义的pre先指向tail.next是因为第一次反转时head的下一个节点应是tail的下一个节点(下一段需要反转的头节点)
比如现在要反转的是1-2这段,本次反转后变成2-1,而1应指向3。
while中的循环就是三个指针法反转链表的操作。
注意:最后要返回的是反转后的链表,这个时候头尾指针应对调返回,也就是头变成尾,尾变成了头。
反转整个链表的操作:
public ListNode reverseKGroup(ListNode head, int k) {
ListNode hair = new ListNode(0,head),tail=null,pre=hair;
while(head!=null){
tail=pre;
for(int i=0;i<k;i++){
tail=tail.next;
if(tail==null) return hair.next;
}
ListNode[] listNode = reverse(head,tail);
head=listNode[0];
tail=listNode[1];
pre.next=head;
head=tail.next;
pre=tail;
}
return hair.next;
}
定义虚拟头节点因为头结点head可能会发生改变。定义pre指向当前需要反转的区域的前一个节点。
循环中先把tail指向需要反转区域的最后一个节点,注意如果tail走到null的时候(题目中说的节点总数不是 k 的整数倍的情况),那么就不需再进行反转了,直接返回即可。
获取到反转完的区域的链表后,head和tail分别进行接收,得到新反转后的区域的head和tail。
然后pre也就是需要反转的区域的前一个节点的下一个节点就是新返回的head。(pre.next=head)
最后head指向下一次需要反转的区域的头( head=tail.next),pre指向下一次需要反转的区域的前一个节点(pre=tail)。
重新回到循环,继续找到下一段反转区域的tail。
最后返回虚拟节点的下一节点即可。