LeetCode第25题-K个一组翻转链表-java实现-图解思路与手撕代码
一、题目描述
给你链表的头节点 head ,每 k 个节点一组进行翻转,请你返回修改后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
示例:
二、解题思路与代码实现
1.解题思路
这道题首先能想到用栈来实现,但是这样做时间复杂度和空间复杂度都比较高,尝试采用尾插法解决问题。
尾插法是指将指定区间内的节点插入到尾部,如下图所示,
输入是[1,2,3,4,5],步长为3,将tail移动步长个单位距离,将head到tail区间内的节点依次插入到tail尾部,最后更新pre,head,tail,cur。
pre
tail head
dummy 1 2 3 4 5
# 我们用tail 移到要翻转的部分最后一个元素
pre head tail
dummy 1 2 3 4 5
cur
# 我们尾插法的意思就是,依次把cur移到tail后面
pre tail head
dummy 2 3 1 4 5
cur
# 依次类推
pre tail head
dummy 3 2 1 4 5
cur
....
2.代码实现
代码如下:
private static ListNode reverseKGroup2(ListNode head, int k) {
ListNode dummy = new ListNode(0, head);
ListNode pre = dummy, cur = pre.next, tail = dummy;
while (true) {
for (int i = 0; i < k; i++) {
if (tail == null || tail.next == null) {
return dummy.next;
}
tail = tail.next;
}
while (cur != tail) {
pre.next=cur.next;
cur.next=tail.next;
tail.next=cur;
cur=pre.next;
}
tail=head;
head=tail.next;
pre=tail;
cur=pre.next;
}
}
需要注意的是每次插入到尾部时更新pre.next,插入结束后需要更新所有指针。
总结
对于链表问题,尾插法不失为一种很有效的解决方法。