LeetCode25. K 个一组翻转链表

题目

分析

类比LeetCode24.两两交换链表种的节点

代码

/**
 * 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),p = dummy;
        dummy->next = head;

        while(1){
            //1. 首先判断有无 k 个节点
            auto q = p;
            for(int i = 0;i < k && q;i++) q = q->next;
            if(!q) break; //没有k个节点就停

            //2. 对k个节点内部反转,就是两两节点换边的方向(k 个节点反转需要k-1次反转边)
            auto a = p->next,b = a->next; 
            for(int i = 0;i < k - 1;i++){
                auto c = b->next;
                b->next = a;
                a = b,b = c;
            }
            
            //注意结束时 a 在 k 个节点最后一个位置，b在k+1节点

            //3.处理边界
            auto c = p->next;
            
            p->next = a,c->next = b;
            p = c;
        }

        return dummy->next;
    }
};

 

时间复杂度O(N)