力扣题目
解题思路
java代码
力扣题目:
给你链表的头节点 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)
额外内存空间的算法解决此问题吗?
解题思路:
解题思路如下:
-
创建哑结点(dummy node):创建一个哑结点作为新链表的头部,这样可以避免处理头部节点的特殊情况。哑结点的下一个节点指向原始链表的头部。
-
定位分组:使用两个指针
pre
和end
来定位每一组的开始和结束位置。pre
指向当前要翻转组的上一个节点,end
指向当前要翻转组的最后一个节点。 -
翻转链表:对于每一组,使用头插法进行局部翻转。首先,将
end.next
设置为null
,然后将pre.next
到end
之间的节点进行翻转。翻转后,将翻转后的链表连接到pre
和end.next
。 -
移动指针:翻转完成后,将
pre
移动到当前组的最后一个节点(翻转后的头节点),end
移动到pre
的位置,然后继续寻找下一组。 -
终止条件:当
end
指向的下一个节点为null
时,表示没有足够的节点进行翻转,循环结束。 -
返回结果:返回哑结点的下一个节点,即新链表的头部。
这个算法的时间复杂度是O(n),因为每个节点都只被访问和修改一次。空间复杂度是O(1),因为除了几个辅助指针外,没有使用额外的空间。
代码中还有一个辅助函数reverse(ListNode head)
,用于翻转链表的一部分。这个函数也是使用头插法进行翻转,将当前节点指向pre
,然后pre
和当前节点都向前移动一位,直到当前节点为null
。
java代码:
package org.example;
public class Leetcode25 {
public static void main(String[] args) {
ListNode head = new ListNode(1);
ListNode node1 = new ListNode(2);
ListNode node2 = new ListNode(3);
ListNode node3 = new ListNode(4);
ListNode node4 = new ListNode(5);
head.next = node1;
node1.next = node2;
node2.next = node3;
node3.next = node4;
int k = 2;
ListNode res = reverseKGroup(head, k);
while (res != null) {
System.out.println(res.val);
res = res.next;
}
}
public static ListNode reverseKGroup(ListNode head, int k) {
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode pre = dummy;
ListNode end = dummy;
while (end.next != null) {
for (int i = 0; i < k && end != null; i++) end = end.next;
if (end == null) break;
ListNode start = pre.next;
ListNode next = end.next;
end.next = null;
pre.next = reverse(start);
start.next = next;
pre = start;
end = pre;
}
return dummy.next;
}
private static ListNode reverse(ListNode head) {
ListNode pre = null;
ListNode curr = head;
while (curr != null) {
ListNode next = curr.next;
curr.next = pre;
pre = curr;
curr = next;
}
return pre;
}
// 作者:房建斌学算法
// 链接:https://leetcode.cn/problems/reverse-nodes-in-k-group/solutions/10416/tu-jie-kge-yi-zu-fan-zhuan-lian-biao-by-user7208t/
// 来源:力扣(LeetCode)
// 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
}