刷题-力扣-25

25. K 个一组翻转链表

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group/
作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。
k 是一个正整数，它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

示例：

给你这个链表：1->2->3->4->5
当 k = 2 时，应当返回: 2->1->4->3->5
当 k = 3 时，应当返回: 3->2->1->4->5

说明：

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

题目分析

1. 根据题目描述，颠倒链表中k个结点
2. 使用栈的后进先出的思想，让链表依次入栈k个到vecNode中
3. 从vecNode中依次从后向前出栈到res的末尾
4. 依次循环直到剩余的结点不够k个
5. 把剩余不够k个的结点添加到res的末尾，res即为所求

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        vector<ListNode*> vecNode;
        ListNode* res = nullptr;
        ListNode* p;
        while (head) {
            int i = 0;
            for (i; i < k; i++) {
                if (head) {
                    vecNode.push_back(head);
                    head = head->next;
                } else break;
            }
            if (vecNode.size() != k) break;
            for (i; i > 0; i--) {
                if (res == nullptr) {
                    res = vecNode[i - 1];
                    p = res;
                }
                else {
                    p->next = vecNode[i - 1];
                    p = p->next;
                    p->next = nullptr;
                }
            }
            vecNode.clear();
        }
        for (int i = 0; i < vecNode.size(); i++) {
            p->next = vecNode[i];
            p = p->next;
            p->next = nullptr;
        }
        return res;
    }
};