LeetCode-025 Reverse Nodes in k-Group

Description

Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.

k is a positive integer and is less than or equal to the length of the linked list. If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.

Example

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

Note

• Only constant extra memory is allowed.
• You may not alter the values in the list’s nodes, only nodes itself may be changed.

Analyse

题意为：将链表中的每k个节点倒序，要求不能更改节点的值和只允许常数的空间复杂度。其实这道题也就是上一道题的增强版，我们只需要实现倒序的操作，然后确定需要倒序的位置就能将这道题套到上一道题中。也就是链表倒序+端点连接。
我们先看看链表怎么倒序吧。（这好像是某年一个大公司的面试题，要求手写代码将一个链表倒序，不能开数组）。
有三种方法：
1、开一个数组，然后将链表放进去，然后再倒序放在一个新链表中，很明显，题目不让我们这样做，而且这样的效率也不高。
2、开三个指针，然后逐个逐个的节点的反转，这种方法是可以的，不过有点需要想象力，三个指针在脑子里面操作的确有点难。
3、固定头节点和第二个节点，然后将后面的节点逐个逐个插在头节点和第二个节点之间，然后让第二个节点连到头节点形成一个环，然后断开环。这种方法是我比较喜欢的方法，过程就像这样：
1->2->3->4
1->3->2->4
1->4->3->2
1->4->3->2->1（接成环）
4->3->2->1（断环）
然后剩下的操作就很像上一道题了，用一个指针first记录需要反转段的第一个节点，用另一个指针second记录需要反转段的最后一个节点，然后进行链表反转。（尾部的节点还是像前一道题一样，需要去记录下来。）

Code

/**
 * 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 *first = head, *second = new ListNode(-1);
		second->next = head;
		ListNode *ans = head, *c = NULL;
		bool finish = false;
		while (!finish) {
			//移动second
			for (int i = 0; i < k; i++) {
				second = second->next;
				if (second == NULL) {
					return ans;
				}
			}
			ListNode *temp = second->next;
			ListNode *p = first->next;
			ListNode *q;
			//反转链表段
			while (p->next != temp) {
				q = p->next;
				p->next = q->next;
				q->next = first->next;
				first->next = q;
			}
			p->next = first;
			second = first->next;
			first->next = temp;
			if (c == NULL) ans = second;
			else c->next = second;
			c = first;
			second = first;
			first = first->next;
		}
		return ans;
	}
};