leetcode 日经贴，Cpp code -Reverse Nodes in k-Group

原创于 2015-05-04 12:25:18 发布 · 311 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一种链表操作算法——K组翻转。该算法实现了一个功能，即每K个一组翻转链表中的节点顺序。文章通过C++代码详细展示了如何实现这一过程，包括定义链表结构、翻转函数的逻辑处理等。

Reverse Nodes in k-Group

/**
 * 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) {
        ListNode *h = NULL, *pre = NULL;
        if (!head || !head->next || k <= 1) {
            return head;
        }
        while (head) {
            int tk = k - 1;
            ListNode *tail = head;
            while (tk && tail) {
                tail = tail->next;
                --tk;
            }
            if (!tail) {
                if (pre) {
                    pre->next = head;
                } else {
                    h = pre = head;
                }
                head = NULL;
            } else {
                ListNode *next = tail->next;
                ListNode *p = head->next;
                head->next = NULL;
                tail = head;
                while (p != next) {
                    ListNode *temp = p->next;
                    p->next = head;
                    head = p;
                    p = temp;
                }
                if (!h) {
                    h = head;
                } else {
                    pre->next = head;
                }
                pre = tail;
                head = next;
            }
        }
        return h;
    }
};