每 k 个节点一组翻转链表
题目25:给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
解题思路:
转自leetocode解法
链接:https://leetcode-cn.com/problems/two-sum/solution/tu-jie-kge-yi-zu-fan-zhuan-lian-biao-by-user7208t/
步骤分解:
- 链表分区为已翻转部分+待翻转部分+未翻转部分
- 每次翻转前,要确定翻转链表的范围,这个必须通过 k 此循环来确定
- 需记录翻转链表前驱和后继,方便翻转完成后把已翻转部分和未翻转部分连接起来
- 初始需要两个变量 pre 和 end,pre 代表待翻转链表的前驱,end 代表待翻转链表的末尾
- 经过k此循环,end 到达末尾,记录待翻转链表的后继 next = end.next
- 翻转链表,然后将三部分链表连接起来,然后重置 pre 和 end 指针,然后进入下一次循环
- 特殊情况,当翻转部分长度不足 k 时,在定位 end 完成后,end==null,已经到达末尾,说明题目已完成,直接返回即可
时间复杂度为 O(n*K)O(n∗K) 最好的情况为 O(n)O(n) 最差的情况未 O(n^2)O(n 2)
空间复杂度为 O(1)O(1) 除了几个必须的节点指针外,我们并没有占用其他空间
/**
* 描述: 每k个节点一组进行反转链表
*
* @author pengjie_yao
* @date 2019/8/6 21:35
*/
public class reverseListNodeK {
/**
* 定义的节点
*/
public static class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}
/**
* 每k个节点进行翻转链表
* @param head
* @param k
* @return
*/
public static ListNode reverseKGroup(ListNode head, int k) {
// 1.判空
if (head == null || k == 0) {
return null;
}
// 2.定义一个头结点的前节点
ListNode preHead = new ListNode(0);
preHead.next = head;
// 3. 定义一个前节点和尾节点,用来保存翻转的前前节点以及尾节点
ListNode pre = preHead;
ListNode end = preHead;
// 4. 遍历链表
while (end.next != null) {
// 5.获取k个节点后的尾节点
for (int i = 0; i < k && end != null; i++) {
end = end.next;
}
// 6.如果已经为空,则直接跳出循环
if (end == null) {
break;
}
// 7.保存前节点,以及尾节点的下一个节点
ListNode start = pre.next;
ListNode next = end.next;
end.next = null;
// 8.进行反转,返回翻转后的节点
ListNode reverseList = reverseNodeList(start);
// 9.拼接头部
pre.next = reverseList;
// 10.拼接尾部,此时反转后的start节点,其实是尾节点
start.next = next;
// 11.重置
pre = start;
end = pre;
}
// 12.返回结果
return preHead.next;
}
/**
* 反转链表
* @param head
* @return
*/
private static ListNode reverseNodeList(ListNode head) {
ListNode current = head;
ListNode pre = null;
while (current != null) {
// 保留下一个节点
ListNode next = current.next;
// 当前节点指向前节点
current.next = pre;
// 前节点成为当前节点
pre = current;
// 让下一个节点成为当前节点
current = next;
}
return pre;
}
public static void main(String[] args) {
ListNode listNode = new ListNode(1);
ListNode listNode1 = new ListNode(2);
ListNode listNode2 = new ListNode(3);
ListNode listNode3 = new ListNode(4);
ListNode listNode4 = new ListNode(5);
ListNode listNode5 = new ListNode(6);
listNode.next = listNode1;
listNode1.next = listNode2;
listNode2.next = listNode3;
listNode3.next = listNode4;
listNode4.next = listNode5;
ListNode resultNode= reverseKGroup(listNode, 3);
while (resultNode != null) {
System.out.println(resultNode.val);
resultNode = resultNode.next;
}
}
}