K 个一组翻转链表
一、原题
给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。
k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。
示例 1：
输入：head = [1,2,3,4,5], k = 2
输出：[2,1,4,3,5]
示例 2：
输入：head = [1,2,3,4,5], k = 3
输出：[3,2,1,4,5]
来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/reverse-nodes-in-k-group
本题需要实现组内反转，如果不够一组，就不进行反转
首先设置链表头，可以先确定是否能找到组内尾节点，找不到直接返回，找到的话，直接设置该节点为头（因为反转后，尾巴就变成了开头），然后进行组内反转，并连接下一组的头节点，到此已经完成了第一步，即设置头，并且完成第一组反转；进行第二组操作，还是先找组内尾结点，然后组内反转，并连接下一组的头节点，进行第三组操作…依次这样操作,直到没有下一组，直接返回
题中使用的链表结构
public static class ListNode {
int val;
ListNode next;
ListNode() {
}
ListNode(int val) {
this.val = val;
}
ListNode(int val, ListNode next) {
this.val = val;
this.next = next;
}
}
代码实现
public class ReverseKGroup {
/**
* 假设链表为1-2-3-4-5，k=2
*/
public static ListNode reverseKGroup(ListNode head, int k) {
if (head == null) {
return null;
}
// statr:1-2-3-4-5
// end:2-3-4-5
ListNode start = head;
ListNode end = getLastNode(head, k);
// 表示没找到符合的一组，不反转，直接返回
if (end == null) {
return head;
}
reverse(start, end);
// statr:1-3-4-5
// end:2-1-3-4-5
// 设置头节点：即第一组反转后的头（原组的尾）
head = end;
// 上一组的尾巴
ListNode lastEnd = start;
while (lastEnd.next != null) {
start = lastEnd.next;
end = getLastNode(start, k);
if (end == null) {
return head;
}
reverse(start, end);
lastEnd.next = end; // 上一组连接上当前组
lastEnd = start;
}
return head;
}
/**
* 找到最后一个节点
*/
private static ListNode getLastNode(ListNode head, int k) {
while (--k > 0 && head != null) {
head = head.next;
}
return head;
}
/**
* 开始和结束节点之前翻转，然后连接下一组节点开头
*/
public static void reverse(ListNode start, ListNode end) {
end = end.next; // 结束节点，该节点不需要反转
ListNode cur = start; // 需要一个索引去遍历，start还是指向原来的头，方便最后链接下一组的头
ListNode pre = null;
ListNode next = null;
while (cur != end) {
next = cur.next;
cur.next = pre;
pre = cur;
cur = next;
}
start.next = end;
}
}
