LeetCode 【25.K个一组反转链表】

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
	ListNode* reverseKGroup(ListNode* head, int k) {
		if (k == 1) return head;
		ListNode* p = head;
		ListNode* q = head;
		ListNode* newHead = head;
		ListNode* pre = head;
		ListNode* pre2 = nullptr;
		while(q != nullptr) {
			int c = k;
			// K个一组反转，一定要是--c，不能是c--：c--跳出循环时c为-1
			while(--c && q->next!=nullptr) q=q->next; 
			// 记录每一段的起点
			pre = p;
			while(c==0) {
				ListNode* p2 = p->next;
				p->next = q->next;
				q->next = p;
				p = p2;
				if (p==q) {
					// 如果上一段存在，则上一段的尾节点连接到本段的头节点
					if (pre2 != nullptr) pre2->next = p;
					// 如果为第一段，则返回头节点
					if (pre2 == nullptr) newHead = p;
					// 将本段的尾节点保存为上一段的尾节点，准备下一次循环
					pre2 = pre;
					p = pre->next;
					q = p;
					break;
				}
			}
			// 如果有余数，则直接返回
			if (c!=0) break;
		}
		return newHead;
	}
};