25. K 个一组翻转链表(非递归实现)

Posted 2021-08-28 issue是fw

LINK

我最讨厌链表了

这个问题如果能额外开空间就是傻瓜题了,可惜leedcode总喜欢玩花的,喜欢$O(1)$空间

于是步骤大概分配下面几步

Ⅰ.从当前组的头指针走$k-1$步判断本组是否满$k$个节点,是否需要翻转.若不满退出循环

Ⅱ.若需要翻转,先保存当前组的头尾指针

Ⅲ.从本组的头节点开始,把每个节点的指针反向.特别的,头指针指向上一组翻转后的尾指针

Ⅳ.保存一下翻转后的尾指针,也就是翻转前的头指针,并存下下一组的头指针迭代

class Solution {
public:
    void rotate(ListNode *head,int k)
    {
        ListNode *pre = head, *cur = head->next, *nxt = nullptr; 
        for(int i=1;i<k;i++)
        {
            nxt = cur->next;
            cur->next = pre, pre = cur, cur = nxt;
        }    
    }
    ListNode* reverseKGroup(ListNode* head, int k) 
    {
        ListNode *now = head;
        for(int i=1;i<k && now!=nullptr ;i++)   now = now->next;
        if( now==nullptr )  return head;
        ListNode *t1 = now->next, *t2 = head, *ans = now;
        rotate( head,k );
        ListNode *begi = t1, *tail = t2;
        while( true )
        {
            ListNode *now = begi;
            for(int i=1;i<k && now!=nullptr ;i++)    now = now->next;
            if( now==nullptr )
            {
            	tail->next = begi;
            	break;
			}
            tail->next = now;
            ListNode *t1 = now->next, *t2 = begi;
            rotate( begi,k );
            begi = t1, tail = t2;
        }
        return ans;
    }
};