关闭

Reverse Nodes in k-Group 链表

258人阅读 评论(0) 收藏 举报
分类:

思路:
链表操作。

从头结点开始,每隔k个节点reverse一次,如果不足k个节点,返回。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
private:
    ListNode *reverse(ListNode *prev, ListNode *begin, ListNode *end) {

        ListNode *end_next = end->next;

        for(ListNode *p = begin, *cur = p->next, *next = cur->next; cur != end_next; p = cur, cur = next, next = next ? next->next : nullptr) {
            cur->next = p;
        }

        begin->next = end_next;
        prev->next = end;
        return begin;
    }

public:
    ListNode* reverseKGroup(ListNode* head, int k) {

        if(head == nullptr || head->next == nullptr || k < 2) return head;

        ListNode dummy(-1);
        dummy.next = head;

        for(ListNode *prev = &dummy, *end = head; end; end = prev->next) {
            for(int i = 1; i < k && end; ++i) {
                end = end->next;
            }
            if(end == nullptr) break;
            prev = reverse(prev, prev->next, end);
        }

        return dummy.next;
    }
};
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    Always Be Coding
    个人资料
    • 访问:851215次
    • 积分:10641
    • 等级:
    • 排名:第1586名
    • 原创:445篇
    • 转载:54篇
    • 译文:0篇
    • 评论:123条
    文章分类