题目
给你链表的头节点 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
解题思路
1.我们使用栈来解决此题,首先我们声明一个栈(stack),以及一个虚拟头节点(dummy)用来连接我们反转后的节点,设置一个p节点用于连接。
2.我们声明一个 count 用于记录遍历节点的数量,设置一个 temp 用于遍历节点。一边遍历一遍将节点放入栈中(stack),当 temp = null 时则表示链表已经遍历结束,当 count = k 时表示我们已经遍历了 K 个一组的节点,我们就需要结束 while 循环
3.下面我们就要确定是哪种情况使得 while 循环结束,如果是 temp = null使得 while 循环结束,那么我们要判断 count 是否等于 k ,因为我们从例题中可以观察到,最后不满足 k 个一组的节点是不会进行翻转的,如果 count 不等于 k ,我们就直接将 head 接在 p 指向的节点后面,如果 count 等于 k ,那么我们就从栈顶一次取出节点连接在 p 后面,直到栈为空。
4.再将 head 节点指向未处理的第一个节点也就是 temp 节点,处理完所有节点后,返回 dummy.next 即可。
代码实现
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
Deque<ListNode> stack = new ArrayDeque<ListNode>();
ListNode dummy = new ListNode(0);
ListNode p = dummy;
while(true){
int count = 0;
ListNode temp = head;
while(temp != null && count < k){
stack.add(temp);
temp = temp.next;
count++;
}
if(count != k){
p.next = head;
break;
}
while(!stack.isEmpty()){
p.next = stack.pollLast();
p = p.next;
}
head = temp;
}
return dummy.next;
}
}
测试结果