LeetCode (25) 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，数字k表示链表中每k个数字进行一次逆转。题目中要求通过改变结点位置实现要求，而不能通过改变结点的value达到目的（改变value显然是一种作弊的行为 o(╯□╰)o）。

解决思路

这道题目还是比较简单的，示意图如下：

将待逆转的结点保存到Stack中，并且保存逆转结点的前一个指针Priv和下一个指针Next，从Priv指针开始依次链接出栈的结点，并将最后出栈的结点的下一个结点设置为Next即可。

需要注意的地方：

• 第一次逆转的时候Priv = NULL，此时需要设置栈顶的指针为链表的head指针。
• 另外对于链表最后不满k个元素的时候，入栈结点个数小于k，是不进行逆转操作的，因此该方法不会影响链表尾部的结点。

代码如下：

/**
 * 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;

        ListNode* h = head;
        ListNode* priv = NULL;
        stack<ListNode*> stackNodes;

        ListNode* l1 = head;
        ListNode* l2 = head;
        int count = 0;
        while (l2)
        {
            if (count != k-1)
            {
                count++;
                stackNodes.push(l2);
                l2 = l2->next;
            }
            else
            {
                l1 = l2;
                l2 = l2->next;
                count = 0;              
                if (priv)
                    priv->next = l1;
                else
                    h = l1;

                while (!stackNodes.empty())
                {
                    l1->next = stackNodes.top();
                    l1 = l1->next;
                    stackNodes.pop();
                }
                priv = l1;
                priv->next = l2;
            }           
        }
        return h;
    }
};