K 个一组翻转链表

Posted 2020-12-22 肉松松松松

给你一个链表，每 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) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if(head==NULL) return NULL;
        if(k==1)return head;

        ListNode *p1=NULL; //头
        ListNode *p2=NULL;
        ListNode *p3=head;
        ListNode *pNext=NULL;
        ListNode *pPre;
        ListNode *ptemp=NULL;
        while(p3)
        {
            p1=p3;
            pPre=p1;
            for(int i=1;i<k;i++)
            {
                if(p3->next==NULL)
                {
                    if(!ptemp)return head;
                    if(pNext) pNext->next=p1;
                    return ptemp;  
                }
                p3=p3->next;
            }
           
            if(!ptemp)ptemp=p3;
            if(pNext)pNext->next=p3;
            p2=p1->next;
            p3=p3->next;
            p1->next=NULL;
            while(p2 && p2!=p3)
            {
                pNext=p2->next;
                p2->next=pPre;
                pPre=p2;
                p2=pNext;
            }
            if(pNext)pNext=p1;
        }    
        return ptemp;
    }
};