题目描述
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
样例描述
输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
思路
- 本题思路与下面这题类似,
Leetcode–Java–24. 两两交换链表中的节点
三个步骤: - 先反转内部k个结点 (这个用的是双指针法,可参考下题)
Leetcode–Java–206. Reverse Linked List - 修改第一个结点(一开始是虚拟结点)的指向
- 修改反转后最后一个结点的指向,指向下一轮的待反转的结点
代码
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
if (head == null || head.next == null || k == 1) return head;
ListNode dummy = new ListNode(-1);
dummy.next = head;
//通过工作指针为空来跳出循环
ListNode p = dummy;
while (true) {
ListNode q = p;
//先看是否后面k个都存在
for (int i = 0; i < k && q != null; i ++ ) {
q = q.next;
}
//没有满足k个都存在
if (q == null) break;
ListNode a = p.next, b = a.next; //两个翻转工作指针
//将k个结点进行内部翻转,需要k - 1次翻转操作
for (int i = 0; i < k - 1; i ++ ) {
//记住b后面的,防止断链
ListNode c = b.next;
b.next = a;
a = b;
b = c;
}
//修改指向,接入到原链表,这里可以手画图对照更直观
ListNode c = p.next;
p.next = a;
c.next = b;
p = c;
}
return dummy.next;
}
}