每日一题-25. K 个一组翻转链表(20200516)

今天是2020年5月16日，星期六。

题目描述

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。

k 是一个正整数，它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

示例：

给你这个链表：1->2->3->4->5

当 k = 2 时，应当返回: 2->1->4->3->5

当 k = 3 时，应当返回: 3->2->1->4->5

说明：

你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

题目思路

本题目是一道链表相关的题目，将节点每k个分为一组，最后剩余的节点个数不足k个不在处理。

链表题目中通常设置一个「虚拟头节点」就可以很好的解决一些问题。本题目我们也设置一个虚拟头结点。在遍历过程中，经过k个节点，就交给单独的方法buildList去处理，这样我们只需要每次遍历，将逆序的过程交给单独的方法去处理即可。

注意：小编的这个代码并不是最优解法，时间上还是消耗比较多的，但是理解起来比较好理解。

参考代码

class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        if (head == null) {
            return null;
        }

        ListNode fast = head;

        // 设置一个虚拟头结点
        ListNode newHead = new ListNode(-1);
        ListNode currentNode = newHead;

        while (true) {
            List<ListNode> listNodes = new ArrayList<>(k);
            for (int i = 1; i <= k; i++) {
                if (fast != null) {
                    ListNode node = new ListNode(fast.val);
                    node.next = fast.next;
                    listNodes.add(node);

                    fast = fast.next;
                } else {
                    break;
                }
            }

            if (fast != null) {
                currentNode = buildList(listNodes, currentNode);
            } else {
                if (listNodes.size() == k) {
                    currentNode = buildList(listNodes, currentNode);
                } else {
                    for (int i = 0; i < listNodes.size(); i++) {
                        currentNode.next = listNodes.get(i);
                        currentNode = currentNode.next;
                    }
                }
                currentNode.next = null;
                break;
            }
        }

        return newHead.next;
    }

    private ListNode buildList(List<ListNode> listNodes, ListNode currentNode) {
        int size = listNodes.size();

        for (int i = size - 1; i >= 0; i--) {
            currentNode.next = listNodes.get(i);
            currentNode = currentNode.next;
        }

        return currentNode;
    }
}