leetcode 25. K 个一组翻转链表 C++实现

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

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

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

 

示例：

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

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

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

 

说明：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
// 分为两步 1.首先判断是否可以翻转
/     如果可以翻转，首先翻转first->kNode这一段
//    然后再翻转kNode->next 到下一个kNode这一段
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if(k<=1 || head == NULL || head->next == NULL) return head;
        ListNode* first = head;
        ListNode* kNode = head;
        for(int i = k; i > 1 && kNode !=NULL; i--){
            kNode = kNode->next;
        }
        if(kNode == NULL) return head; //判断是否可以翻转
        kNode = head;
        ListNode* result = new ListNode(1);
        ListNode* pre = result;
        while(kNode!= NULL) {  //遍历，知道链表末尾
            int num = k;
            while(num >1 && kNode!=NULL) { // 大于1是因为要保留第K个节点
               kNode = kNode->next;
               num--;
            } 
            if(kNode == NULL) break;//为空说明不够k个不能进行发转
            ListNode* tp = first;
             ListNode* tp2 = kNode->next;
            pre->next = getReverseData(first, kNode);
            pre = tp;  // 更新pre因为再getReverse中已经改变了first
            first = tp2;
            kNode = tp2; // 更新kNode，因为再getReverse中已经改变了kNode的next
       }
        return result->next;
    }
    ListNode* getReverseData(ListNode* first, ListNode* kNode) {
        ListNode* next = first->next;
        first->next = kNode->next;  //用于翻转first到kNode这一段
        while(next!=NULL && next != kNode) {
            ListNode* tmp = next->next;
            next->next = first;
            first= next;
            next = tmp;
        }
        if(next !=NULL) next->next = first;
        return next;
    }
};