精选力扣500题第5题 LeetCode 25. K个一组翻转链表 c++详细题解

Posted 2021-05-21 林深时不见鹿

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了精选力扣500题第5题 LeetCode 25. K个一组翻转链表 c++详细题解相关的知识，希望对你有一定的参考价值。

1、题目

给你一个链表，每 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]

示例 3：

输入：head = [1,2,3,4,5], k = 1
输出：[1,2,3,4,5]

示例 4：

输入：head = [1], k = 1
输出：[1]

提示：

列表中节点的数量在范围 sz 内
1 <= sz <= 5000
0 <= Node.val <= 1000
1 <= k <= sz

2、思路：

在这里插入图片描述
具体操作，如图所示，黑色表示当前原始状况，红色表示操作状况

1、将后k个元素进行倒序操作，共操作k - 1次，详细操作看代码（注意：需要确保有k个元素才操作）
2、将后k个元素倒序后的序列的尾部指向后k + 1的元素
3、将p.next指向倒序后的序列的头部
4、更新p指针的位置，即指向倒序后的序列的尾部

3、代码

/**
 * 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) {
        auto dummy =  new ListNode(-1);
        dummy->next = head;
        for(auto p = dummy; ;)
        {
            auto q = p;
            for(int i = 0;  i < k && q; i++) q = q->next;
            if(!q) break;
            auto a = p->next,b = a->next;
            for(int i = 0; i < k - 1; i++)
            {
                auto next = b->next;
                b->next = a;
                a = b, b = next;
            }
            auto c = p->next;
            c->next = b,p->next = a;
            p = c;
        }
        return dummy->next;
    }
};