LeetCode25 K个一组翻转链表

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

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

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

示例：

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

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

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

思路比较简单，现在前面设置一个dummyhead，然后连续翻转长度为k的子链表，注意子链表和头尾的链接。

这里采用的比较复杂的方法，一次传入了4个参数：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if (head == nullptr)
            return nullptr;
        if (k < 2)
            return head;
        ListNode* dummyhead = new ListNode(-1);
        dummyhead->next = head;
        ListNode* phead = head, * pbefore = dummyhead, * pafter = nullptr, * ptail = nullptr;
        int count = 0;
        ListNode* cur = head;
        while (cur != nullptr) {
            ++count;
            if (count == k) {
                ptail = cur;
                pafter = ptail->next;
                reverse_list(pbefore, phead, ptail, pafter);
                pbefore = phead;
                phead = pafter;
                count = 0;
                cur = phead;
            }
            else {
                cur = cur->next;
            }
        }
        return dummyhead->next;
    }

    void reverse_list(ListNode* before, ListNode* head, ListNode* tail, ListNode* after) {
        ListNode* cur = head, * save = head->next, * last = after;
        before->next = tail;
        while (cur != after) {
            cur->next = last;
            last = cur;
            cur = save;
            if(cur!=nullptr)
                save = cur->next;
        }
        return ;
    }
};

答案给的很简单，使用了pair返回新的头和尾

class Solution {
public:
    // 翻转一个子链表，并且返回新的头与尾
    pair<ListNode*, ListNode*> myReverse(ListNode* head, ListNode* tail) {
        ListNode* prev = tail->next;
        ListNode* p = head;
        while (prev != tail) {
            ListNode* nex = p->next;
            p->next = prev;
            prev = p;
            p = nex;
        }
        return {tail, head};
    }

    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode* hair = new ListNode(0);
        hair->next = head;
        ListNode* pre = hair;

        while (head) {
            ListNode* tail = pre;
            // 查看剩余部分长度是否大于等于 k
            for (int i = 0; i < k; ++i) {
                tail = tail->next;
                if (!tail) {
                    return hair->next;
                }
            }
            ListNode* nex = tail->next;
            // 这里是 C++17 的写法，也可以写成
            // pair<ListNode*, ListNode*> result = myReverse(head, tail);
            // head = result.first;
            // tail = result.second;
            tie(head, tail) = myReverse(head, tail);
            // 把子链表重新接回原链表
            pre->next = head;
            tail->next = nex;
            pre = tail;
            head = tail->next;
        }

        return hair->next;
    }
};

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group/solution/k-ge-yi-zu-fan-zhuan-lian-biao-by-leetcode-solutio/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。