LeetCode 25.Reverse Nodes in k-Group K 个一组翻转链表

题目描述

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

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

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

示例 :

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

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

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

说明 :

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

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

思路

先进行遍历，每k个进行一次遍历，找到需要翻转的节点的开始节点和结束节点；

然后使用头插法对需要反转的节点进行反转。[begin, end]

c++

/**
 * 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 || head->next == nullptr || k < 2) return head;
        ListNode dummy(-1);
        dummy.next = head;
        ListNode *prev = &dummy, *end = head;;
        while(end){
            for(int i = 0; i < k - 1 && end; ++i){
                end = end->next;  
            }
            if(end == nullptr)  break;  // 不够整除
            // 进行反转，可以写成一个函数
            // prev = reverse(prev, prev->next, end);
            ListNode *begin = prev->next, *end_next = end->next, *curr=begin->next;
            while(curr != end_next){
                begin->next = curr->next;
                curr->next = prev->next;
                prev->next = curr;
                curr = begin->next;
            }
            prev = begin;
            end = prev->next;
        }
        return dummy.next;
    }
};