题目主要信息:
- 给定一个链表,从头开始每k个作为一组,将每组的链表节点翻转
- 组与组之间的位置不变
- 如果最后链表末尾剩余不足k个元素,则不翻转,直接放在最后
解题思路:
递归:如果这个链表有n个分组可以反转,我们首先对第一个分组反转,接下来将剩余n−1个分组反转后的结果接在第一组后面就行了,那这剩余的n−1组就是一个子问题。我们来看看递归的三段式模版:
- 终止条件: 当进行到最后一个分组,即不足k次遍历到链表尾(0次也算),就将剩余的部分直接返回。
- 返回值:每一级要返回的就是翻转后的这一分组的头,以及连接好它后面所有翻转好的分组链表。
- 本级任务:对于每个子问题,先遍历k次,找到该组结尾在哪里,然后从这一组开头遍历到结尾,依次翻转,结尾就可以作为下一个分组的开头,而先前指向开头的元素已经跑到了这一分组的最后,可以用它来连接它后面的子问题,即后面分组的头。
具体做法:
- step 1:每次从进入函数的头节点优先遍历链表k次,分出一组,若是后续不足k个节点,不用反转直接返回头。
- step2:从进入函数的头节点开始,依次反转接下来的一组链表,反转过程同。
- step3:这一组经过反转后,原来的头变成了尾,后面接下一组的反转结果,下一组采用上述递归继续。
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* }
*/
public class Solution {
/**
*
* @param head ListNode类
* @param k int整型
* @return ListNode类
*/
public ListNode reverseKGroup (ListNode head, int k) {
// write code here
if(head == null || head.next == null)
return head;
ListNode temp = head;
for(int i = 0; i < k; i++){
if(temp == null)
return head;
temp = temp.next;
}
ListNode newHead = reverse(head,temp);
head.next = reverseKGroup(temp,k) ;
return newHead;
}
private ListNode reverse(ListNode head,ListNode temp){
ListNode pre = null;
ListNode next = null;
while(head != temp){
next = head.next;
head.next = pre;
pre = head;
head = next;
}
return pre;
}
}