题解 | 链表中的节点每k个一组翻转

链表中的节点每k个一组翻转

https://www.nowcoder.com/practice/b49c3dc907814e9bbfa8437c251b028e

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverse(ListNode* pre, int k){
        // 给出一组的前一个节点,然后翻转该组,返回结果指向翻转后的头结点。
        ListNode* cur = pre->next;
        for(int m=1; m<k; m++){
            ListNode* temp = cur->next;
            cur->next = temp->next;
            temp->next = pre->next;
            pre->next = temp;
        }
        return pre->next;
    }

    ListNode* reverseKGroup(ListNode* head, int k) {
        // write code here
        if(k <= 1) return head;
        if(head == NULL) return head;
        ListNode* dummy = new ListNode(0);
        dummy->next = head;
        ListNode* prev = dummy;
        ListNode* cur = head;
        int len = 0;
        while(cur){
            cur = cur->next;
            len++;
        }
        int s = len / k;
        for(int i = 0; i < s; i++){
            if (i == 0){
                dummy->next = reverse(prev, k);
                prev->next = dummy->next;
            }
            else{
                prev->next = reverse(prev, k);
            }
            for(int j = 0; j < k; j++){
                prev = prev->next;
            }
        }
        return dummy->next;
    }
};

类似BM2的解题思路。构造一个reverse函数快速实现组内的翻转,然后添加k组的思路即可。

全部评论

相关推荐

04-08 23:37
已编辑
东华大学 结构工程师
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务