K 个一组翻转链表

题目描述

来源于 https://leetcode-cn.com/

给出一个链表,每 个节点一组进行翻转,并返回翻转后的链表。

是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 的整数倍,那么将最后剩余节点保持原有顺序。

示例 :

给定这个链表:1->2->3->4->5

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

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

说明 :

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

解法:

先把前 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) {
        int i = 1;
        ListNode* p = head;
        while(i < k && p != nullptr){
            p = p->next;
            i++;
        }
        if(i == k && p != nullptr){
            ListNode *tail = head;
            ListNode* next = p->next;
            p->next = nullptr;
            head = reverse(head);
            tail->next = reverseKGroup(next, k);
            return head;
        }else{
            return head;
        }
    }
    ListNode* reverse(ListNode* head){
        ListNode* p = head;
        head = nullptr;
        while(p){
            ListNode* next = p->next;
            p->next = head;
            head = p;
            p = next;
        }
        return head;
    }
};