题目:
给你链表的头节点 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]
提示:
链表中的节点数目为 n
1 <= k <= n <= 5000
0 <= Node.val <= 1000
进阶:你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗?
解题思路:首先判断链表的节点总数是否小于k,若是,则直接返回原链表,针对每个k个节点的子链表,进行翻转操作。定义一个指针cur指向当前子链表的头节点,以及一个指针prev指向上一个子链表的尾节点,用于连接翻转后的子链表;使用一个计数器count来记录当前节点的位置,每遍历一个节点,count加1;当count等于k时,说明已遍历到一个子链表的尾部;将cur的next指针保存为next_node,用于指向下一个子链表的头节点;对当前的子链表进行翻转操作,得到翻转后的子链表new_head;对于第一个子链表,将链表的头节点指向new_head;对于其他子链表,将上一个子链表的尾节点的next指针指向new_head;将prev指向当前子链表的尾节点,将cur指向next_node,然后将count重置为1;遍历完成后,返回修改后的链表。
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
// 判断链表长度是否小于k
int count = 0;
ListNode node = head;
while (count < k && node != null) {
node = node.next;
count++;
}
// 如果链表长度小于k,则不需要翻转,直接返回原链表
if (count < k) {
return head;
}
// 定义指针cur和prev
ListNode cur = head;
ListNode prev = null;
// 翻转当前子链表
count = 0;
while (count < k && cur != null) {
ListNode nextNode = cur.next;
cur.next = prev;
prev = cur;
cur = nextNode;
count++;
}
// 递归翻转下一个子链表,并将翻转后的子链表连接到当前翻转后的子链表的尾部
if (cur != null) {
head.next = reverseKGroup(cur, k);
}
return prev;
}
}