LeetCode题解： Reverse Nodes in k-Group

最新推荐文章于 2024-11-11 20:59:17 发布

MagiSu

最新推荐文章于 2024-11-11 20:59:17 发布

阅读量1.1k

点赞数

分类专栏：数据结构和算法文章标签： leetcode

本文链接：https://blog.csdn.net/MagiSu/article/details/16886139

版权

数据结构和算法专栏收录该内容

126 篇文章 0 订阅

订阅专栏

这篇博客主要介绍了如何解决LeetCode上的'k组反转链表'问题，要求在不改变节点值的情况下，将链表中的节点按k个一组进行反转。如果节点数不是k的倍数，剩余的节点保持原样。解决方案需要只使用常量级的额外内存。

摘要由CSDN通过智能技术生成

Reverse Nodes in k-Group

Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.

If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.

You may not alter the values in the nodes, only nodes itself may be changed.

Only constant memory is allowed.

For example,
Given this linked list: 1->2->3->4->5

For k = 2, you should return: 2->1->4->3->5

For k = 3, you should return: 3->2->1->4->5

思路：

首先要注意处理k=1的情况。在k不等于1的时候，每次取k个结点到缓冲中去，然后更新除第一个结点之外其他结点的next指针。第一个结点的next指针指向下一个尚未处理的结点。然后继续处理。如果之后又读取了k个链表结点，那么就要更新之前的next指针。

题解：

/**
 * 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 == nullptr)
            return nullptr;
            
        if (k == 1)
            return head;
            
        vector<ListNode*> buffer (k, nullptr);
        
        ListNode* last = nullptr;
        ListNode* iter = head;
        bool first_time = true;
        for (;;)
        {
            bool terminating = false;
            
            for (int i = 0; i < k; ++i)
            {
                if (iter == nullptr)
                {
                    terminating = true;
                    break;
                }
                buffer[i] = iter;
                iter = iter->next;
            }
            
            if (terminating)
                break;
            
            for (int i = k - 1; i > 0; --i)
                buffer[i]->next = buffer[i - 1];
                
            if (last != nullptr)
                last->next = buffer.back();
    
            last = buffer.front();
            last->next = iter;
    
            if (first_time) // update head
            {
                head = buffer.back();
                first_time = false;
            }
        }
        return head;
    }
};