25 reverse-nodes-in-k-group

这道题的大意是说,给一个链表,再给一个整型数字K,要实现这个链表每K个数进行一次反转,整条链表反转的题目比较熟悉,但是,一个链表分成好几组进行反转其实大体思路是一样的,但是每组反转结束之后,应该如何变化指针进行下一次反转,这是这道题的重点。

按照以前链表反转的思路,我们首先要创建一个头节点,用来记录链表头节点的位置。

其次,设置计数值,将每组的头尾节点分别记录,这块要注意,最好要记录每组头节点的前一个节点,因为每组反转后,每组原来的最后一个节点要变为第一个节点,也就是要让上一组的最后一个节点指向它,所以,此处要记录的是每组第一个元素的前一个元素的地址: dummy -> 1 -> 2 -> 3 -> 4 ->NULL,K=2,第一次记录dummy和2,第二次记录2和4,从而实现各组反转。这块想通了代码也就比较简单了。

以下是我的代码:

/**
 * 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<2) {
            return head;
        }
        ListNode* dummy = new ListNode(0);
        ListNode* prev = dummy, *cur = head;
        dummy->next = head;
        int count = 1;        
        while (cur) {
            if (count%k) {
                cur = cur->next;
            } else {
                prev = reverseKNodes(prev, cur->next);
                cur = prev->next;
            }
            ++count;
        }
        return dummy->next;
    }
    
    ListNode* reverseKNodes(ListNode* prev, ListNode* next) {
        ListNode* head = prev->next;
        ListNode* cur = head->next;
        while (cur != next) {
            head->next = cur->next;
            cur->next = prev->next;
            prev->next = cur;
            cur = head->next;
        }
        return head;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_29592167/article/details/83069951