K个一组反转链表

K个一组反转链表

描述

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

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

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

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

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

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

思路和代码

每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) {
        ListNode *dummy = new ListNode(0);
        dummy->next = head;
        
        ListNode *pre = dummy;
        ListNode *end = dummy;
        
        while(end->next!=NULL)
        {
            for(int i=0; i<k && end!=NULL; i++)
                end = end->next;
            if(end == NULL)
                break;
            ListNode *start = pre->next;
            ListNode *next = end->next;
            end->next = NULL;
            pre->next = reverse(start);
            start->next = next;
            pre = start;
            end = start;
        }
        return dummy->next;
    }
    
    ListNode* reverse(ListNode *head)
    {
        ListNode *pre = NULL;
        ListNode *cur = head;
        while(cur!=NULL)
        {
            ListNode *next = cur->next;
            cur->next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }
};