25. K 个一组翻转链表

https://leetcode-cn.com/problems/reverse-nodes-in-k-group/
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。

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

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

进阶:

你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

我的方法一:递归 O(n) O(n/k)

步骤

  1. 递归实现,每次反转当前k个节点,然后递归反转后面的节点

初始条件

  1. 记录前面已反转组的最后一个节点pre_tail,因为当前节点反转后,要连接到pre_tail->next
    1.1 一个优化点使用dummy,这样后续所有的递归不需要考虑pre_tail是否为空,而且最终的结果就是dummy->next
  2. 当前节点组的头节点,注意pre_tail和当前的head一定要断开,即pre_tail->next=null

边界条件

  1. 当前节点组的数量小于k

复杂度

时间:O(n),遍历了一遍所有节点
空间:O(n/k),因为递归调用n/k次,所以会产生n/k的栈占用,所以是O(n/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) {
        ListNode dummy;
        reverseKGroup(&dummy, head, k);
        return dummy.next;
    }

private:
    void reverseKGroup(ListNode* pre_tail, ListNode* head, int k) {
        ListNode* next_head = nullptr;
        ListNode* cur = head;
        int count = 0;
        while(cur) {
            count++;
            if(count == k) {
                next_head = cur->next;
                cur->next = nullptr;
                break;
            }
            cur = cur->next;
        }

        if(count < k) {
            pre_tail->next = head;
            return;
        }

        //reverse
        {
            ListNode* next = nullptr;
            ListNode* pre = nullptr;
            ListNode* cur = head;
            
            while(cur) {
                next = cur->next;
                cur->next = pre;
                pre = cur;
                cur = next;
            }

            pre_tail->next = pre;
        }
        
        reverseKGroup(head, next_head, k);
    }
};

你可能感兴趣的:(25. K 个一组翻转链表)