牛客网高频算法题系列-BM3-链表中的节点每k个一组翻转
题目描述
将给出的链表中的节点每 k 个一组翻转，返回翻转后的链表如果链表中的节点数不是 k 的倍数，将最后剩下的节点保持原样你不能更改节点中的值，只能更改节点本身。
原题目见：BM3 链表中的节点每k个一组翻转
解法一：递归
使用递归的方式处理每k个一组的结点，具体处理方式如下：
- 首先用tail记录每次翻转的尾部，就是第k+1个结点，从头结点开始，遍历链表，找到第k+1个结点，如果当前结点数不够k个，则不用反转直接返回；
- 然后从链表头开始遍历，翻转结点并记录反转后的表头pre和表尾结点head；
- 然后递归处理从tail开始的后面的链表结点，并将head指向递归处理后的链表；
- 返回新的表头即为反转后的新的链表的表头结点。
代码
public class Bm003 {
/**
* 递归
*
* @param head ListNode类
* @param k int整型
* @return ListNode类
*/
public static ListNode reverseKGroup(ListNode head, int k) {
// 找到每次翻转的尾部
ListNode tail = head;
// 遍历k次到尾部
for (int i = 0; i < k; i++) {
// 如果不足k到了链表尾部，直接返回，不翻转
if (tail == null) {
return head;
}
tail = tail.next;
}
// 翻转时需要的前序和当前节点
ListNode pre = null, cur = head;
// 在到达当前段尾部节点前
while (cur != tail) {
ListNode temp = cur.next;
cur.next = pre;
pre = cur;
cur = temp;
}
// 当前尾部指向下一段要翻转的链表
head.next = reverseKGroup(tail, k);
return pre;
}
public static void main(String[] args) {
ListNode head = ListNode.testCase2();
System.out.println("翻转之前");
ListNode.print(head);
// 当k=2时
ListNode newHead = reverseKGroup(head, 2);
System.out.println("当k=2时，翻转之后");
ListNode.print(newHead);
ListNode head1 = ListNode.testCase2();
// 当k=3时
ListNode newHead2 = reverseKGroup(head1, 3);
System.out.println("当k=3时，翻转之后");
ListNode.print(newHead2);
}
}
相信坚持的力量！
