leetcode 25. K 个一组翻转链表

K 个一组翻转链表

 

给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。

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

你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

示例 1：

输入：head = [1,2,3,4,5], k = 2
输出：[2,1,4,3,5]

示例 2：

输入：head = [1,2,3,4,5], k = 3
输出：[3,2,1,4,5]

提示：

• 链表中的节点数目为 n
• 1 <= k <= n <= 5000
• 0 <= Node.val <= 1000

进阶：你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗？

思路：

分段逆转k个节点

逆转后，第一个节点变成k个节点中的最后一个节点，然后递归调用reverseKGroup函数；
需要注意的是当链表中节点不足k个节点时不逆转，所以加一个判断，当节点不足k个时将逆转后的节点再你转一遍。

struct ListNode*  reverselist(struct ListNode* head, int k){

    if(head == NULL)
        return head;
    struct ListNode* ans = malloc(sizeof(struct ListNode));
    struct ListNode* tmp;
    ans->next = head;
    while(head->next != NULL){
        if((--k) == 0)
            break;
        tmp = head->next;
        head ->next = tmp->next;
        tmp->next = ans->next;
        ans->next = tmp;
    }
  //  show(ans->next);
    head = ans->next;
    if(k!=0 && k!=1){
        while(head->next != NULL){
            tmp = head->next;
            head ->next = tmp->next;
            tmp->next = ans->next;
            ans->next = tmp;
        }
    }
    //printf("k == %d\n", k);
    return  ans->next;
}

struct ListNode* reverseKGroup(struct ListNode* head, int k){
    if(head == NULL)
        return NULL;
    struct ListNode* tmp = head;
    struct ListNode* ans = malloc(sizeof(struct ListNode));
    ans->next = reverselist(head, k);
    if(tmp->next != NULL)
        tmp->next = reverseKGroup(tmp->next, k);
    return ans->next;
}