题目描述:
给你一个链表,每 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]
解题思路:
方法一:迭代
使用指针 pre 和 cur 来反转每个 k 个节点,首先反转前 k 个节点,然后链接反转后的头节点和下一组反转后的头节点。
反转链表的操作可以参考力扣206. 反转链表。
时间复杂度:O(N),其中 N 是链表的节点数量。
空间复杂度:O(1)
代码实现(迭代):
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode pre = dummy;
ListNode cur = head;
int length = 0;
while (head != null) {
length++;
head = head.next;
}
for (int i = 0; i < length / k; i++) {
for (int j = 1; j < k; j++) {
ListNode next = cur.next;
cur.next = next.next;
next.next = pre.next;
pre.next = next;
}
pre = cur;
cur = pre.next;
}
return dummy.next;
}
}
方法二:递归
首先判断剩余的节点数量是否大于等于 k,若不满足则不反转直接返回头节点。
对于每一次的 k 个节点,先找到待反转的第 k+1 个节点,然后将当前 k 个节点进行翻转,并将翻转后的头节点返回。
对于翻转 k 个节点的操作,可以参考力扣206. 反转链表。
时间复杂度:O(N)
空间复杂度:O(N/k),其中 N 是链表的节点数量。
代码实现(递归):
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
ListNode cur = head;
int count = 0;
while (cur != null && count < k) {
cur = cur.next;
count++;
}
if (count == k) {
cur = reverseKGroup(cur, k);
while (count > 0) {
ListNode next = head.next;
head.next = cur;
cur = head;
head = next;
count--;
}
head = cur;
}
return head;
}
}
时间复杂度分析:
每次递归操作需要遍历 k 个节点,因此总的递归操作次数为链表长度除以 k,所以时间复杂度为 O(N)。
空间复杂度分析:
每次递归操作使用了 O(k) 的栈空间,因此总的递归操作次数为链表长度除以 k,所以空间复杂度为 O(N/k)。