Leetcode-每日一题【25.k个一组翻转链表】

题目

给你链表的头节点 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;
}
}

测试结果