leetcode25 K 个一组翻转链表

题目

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

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

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

示例 :

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

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

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

说明 :

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

思路

不算难，比较烦，逐k个头插法逆置，注意不满k个返回原始顺序

/**
 * 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 || !head->next) return head;
        ListNode* dummy_head = new ListNode(-1);
        ListNode* dummy_pre = new ListNode(-1);
        ListNode *pre = dummy_head,*cur = head,*cur_pre = dummy_pre,*tmp,*begin,*end;
        int cnt = 0;
        while(cur){
            begin = cur;
            end = cur;
            // 获得k个
            while(cnt < k && end){
                end = end->next;
                cnt += 1;
            }
            // 满足条件则翻转
            if(cnt == k){
                // 头插法逆置
                while(cur != end){
                    tmp = cur->next;
                    cur->next = cur_pre->next;
                    cur_pre->next = cur;
                    cur = tmp;
                }
                // 更新pre，pre更新为k个节点的尾结点，也就是begin
                pre->next = cur_pre->next;
                pre = begin;
                // 重置，继续循环
                cnt = 0;
                cur_pre->next = NULL;
            }else{
                // 不满足条件，直接连接，跳出循环
                pre->next = begin;
                break;
            }
        }
        return dummy_head->next;
    }
};