leetcode25. K 个一组翻转链表
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明:
- 你的算法只能使用常数的额外空间。
- 你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
基本思路
应用链表转置的方法,加上一些细节。
- 第一个是要存储转置之前的首节点的前一个,和尾节点的后一个。比如 [1,2,3,4],反转 [2,3] ,那么就要用start和end存储1和4,这样才能把链表连接起来。
- 第二个要在每次反转之前先判断剩下的长度够不够翻转。这个我没看到优化方法,官方题解也是先查找长度,在翻转的。
java代码。
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
ListNode nHead = new ListNode(0);//虚拟头结点
ListNode tmp = head, start = nHead, end = nHead,tail = nHead;
//start和end记录首尾节点,只不过end记录的是上述例子的4,tail记录的是3
while (tmp != null) {
ListNode t = tmp;
int flag = 1;
//flag判断剩余节点够不够翻转
for (int i = 0; i < k; i++) {
t = t.next;
if (t == null && i != k - 1) {
flag = 0;
break;
}
}
if (flag == 0) {
break;
}
//下面这个连接我觉得写得有点乱,不同人裂解方法可能不一样
//目的就是把翻转之后的链表插回原来的链表,所以需要翻转链表的头结点的前一个,尾节点的后一个,
//为了不用特殊判断头结点,在一开始加了一个虚拟的头结点
start = tail;
end = t;
ListNode n = reverse(tmp,k);
start.next = n;
tmp.next = end;
tail = tmp;
tmp = end;
}
return nHead.next;
}
public ListNode reverse(ListNode head, int length) {
if (head == null) return null;
//返回头结点,当前链表节点长度为length
ListNode nhead = head, next = head.next, cur;
for (int i = 1; i < length; i++) {
cur = next;
next = cur.next;
cur.next = nhead;
nhead = cur;
}
head.next = null;
return nhead;
}
public void show(ListNode head) {
ListNode tmp = head;
while (tmp!=null) {
System.out.print(tmp.val+" ");
tmp = tmp.next;
}
System.out.println();
}
}
leetcode 100/100