K个一组翻转链表—leetcode25

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

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

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

示例：

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

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

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

说明：

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

/**
 * 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(k==1)
            return head;
        ListNode* tempHead = head;
        int length = 0;
        while(tempHead!=NULL){
            length++;
            tempHead = tempHead->next;
        }
        tempHead = head;
        vector vec;
        int index = 0;
        ListNode* temp = NULL;
        while(index+k<=length){
            for(int i=0;inext;
            }
            temp = tempHead->next;
            tempHead->next = NULL;
            tempHead = temp;
            index+=k;
        }

        if(!vec.empty()){
            int len = vec.size();
            ListNode* node = NULL;
            ListNode* result = NULL;

            if(len>1){
                for(int i=0;inext!=NULL)
                        node = node->next;
                    node->next = reverse(vec[i+1]);
                }
            }else{
                node = reverse(vec[0]);
                result = node;
            }
            while(node->next!=NULL)
                node = node->next;
            node->next = tempHead;
            return result;
        }
        return head;
    }

    ListNode* reverse(ListNode* head){
        ListNode* t = head;
        ListNode* newHead = NULL;
        while(t!=NULL){
            ListNode* temp = t->next;
            t->next = newHead;
            newHead = t;
            t = temp;
        }
        return newHead;
    }
};