[leetcode] 25. Reverse Nodes in k-Group

题意

如例子所示，给定一个链表，以k个元素作为一组，将它们倒置，若剩余不足k个，保持不变。
Given this linked list: 1->2->3->4->5

For k = 2, you should return: 2->1->4->3->5

For k = 3, you should return: 3->2->1->4->5
要求：额外的空间是常数级并且不可改变结点元素

题解

因为不可更改结点元素，只能更改结点自身的位置，就，就不能使用交换元素的方法来解题了(或许可以解，但不符合也没有意义)

一种简单的想法是提取出k个链表，对它进行倒置操作,实际上也是这样做的。

这里可以利用递归，取出k个结点，剩余的链表进行一次调用就可以得到调换后的链表头next_head，将这k个结点倒置后另链表尾的下一个结点为next_head即可，注意讨论特殊情况(当传入链表不足k个结点)。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

static int x=[](){
    std::ios::sync_with_stdio(false);
    cin.tie(nullptr);
    return 0;
}(); 

class Solution {
    
    pair<ListNode*, ListNode*> reverse(ListNode* head, ListNode* rail){
    	ListNode* temp = nullptr, *temp2 = head->next;
        pair<ListNode*, ListNode*> r = make_pair(nullptr, head);
    	while(head != rail){
    		head->next = temp;
    		temp = head;
    		head = temp2;
    		temp2 = temp2->next;
		}
		rail->next = temp;
		r.first = rail;
		return r;
	}
    
    
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
    	if(head == nullptr) return nullptr;
        ListNode* begin = head, *end = head;
        int i;
        for(i = 1; i < k; ++i){
            if(end->next){
                end = end->next;
            }
            else{
            	break;
			}
        }
        if(i != k) return head;
        ListNode* next = reverseKGroup(end->next, k);
        pair<ListNode*, ListNode*> res = reverse(begin, end);
        res.second->next = next;
        return res.first;
    }
};