Loading

leetcode#25 Reverse Nodes in k-Group

给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。

k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。

示例 :

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

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

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

说明 :

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

//如果写成递归,应该会比较简练,但不是常数空间

class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if(!head) return head;
        int count=k;
        auto probe=head;
        while(k--&&probe) probe=probe->next;
        if(k>-1) return head;
        ListNode* newhead=nullptr,newtail=nullptr;
        k=count;
        while(k--)
        {
            if(head)
            {
                auto copy=head->next;
                if(!newtail) newtail=head;
                head->next=newhead;                
                newhead=head;
                head=copy;
            }
        }
        newtail->next=reverseKGroup(head,count);
        return newhead;        
    }
};

 

posted @ 2018-09-29 13:19  老鼠阿尔吉侬  阅读(128)  评论(0编辑  收藏  举报