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) {
        auto dummy=new ListNode(-1);
        dummy->next=head;
        for(auto p=dummy;;){
            auto q=p;
            for(int i=0;i<k&&q;i++) q=q->next;
            if(!q) break;
            auto a=p->next,b=a->next;
            for(int i=1;i<=k-1;i++){
                auto cur=b->next;
                b->next=a;
                a=b,b=cur;
            }
            auto c=p->next;
            p->next=a;
            c->next=b;
            p=c;
        }
        return dummy->next;
    }
};