题解 | #牛群的重新分组# 链表模拟题

知识点

链表模拟

题意分析

以k个结点为一组翻转链表, 剩下的不足k个位置不翻转

翻转链表可以用栈来维护, 建立虚拟头结点来避免头结点的麻烦问题

最后不足k个的时候不需要翻转, 因此把栈全部弹出, 把最后弹出的元素接在结果链表的末尾

时间复杂度

只遍历链表若干次 和链表长度成正比

AC code(C++)

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverseKGroup(ListNode* head, int k) {
        // write code here
        auto dummy = new ListNode(-1);
        stack<ListNode*> stk;
        auto p = head, last = dummy;
        while (p) {
            for (int i = 0; i < k and p; i ++) {
                stk.push(p);
                p = p->next;
            }
            if (stk.size() < k) {
                while (stk.size()) {
                    last->next = stk.top();
                    stk.pop();
                }
                break;
            }
            while (stk.size()) {
                auto t = stk.top();
                stk.pop();
                last->next = t;
                last = t;
            }
            last->next = nullptr;
        }
        return dummy->next;
    }
};