leetcode25. K 个一组翻转链表

题目妙速

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

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

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

 

示例

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

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

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

说明 :

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

 

思路

其实这题就是原地反转链表的加强版而已。

 

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if (!head || k == 1) return head;
        auto tmp = head;
        int length = 0;
        while (tmp){
            length++;
            tmp = tmp->next;
        }
        int num = length / k;
        return dfs(head, 0, num, k);
        
    }
    
    ListNode* dfs(ListNode* head, int cnt, int num, int k){
        if (num == cnt) return head;
        auto end = head, tmp = head->next;
        for (int i = 1; i < k; i++){
            auto next = tmp->next;
            tmp->next = head;
            head = tmp;
            tmp = next;
        }
        end->next = dfs(tmp, cnt+1, num, k);
        return head; 
    }
};