LeetCode 25. K 个一组翻转链表

给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。

k 是一个正整数,它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。

 

示例:

给你这个链表:1->2->3->4->5

当 k = 2 时,应当返回: 2->1->4->3->5

当 k = 3 时,应当返回: 3->2->1->4->5

 

说明:

你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

 

/**
 * 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 == NULL)
            return head;
        
        ListNode* dummy = new ListNode(0);
        dummy->next = head;

        ListNode* prior = dummy, *phead = dummy, *temp, *prhead, *prtail;
        while (true)
        {
            for (int i = 0; i < k && prior; ++i)
                prior = prior->next;
            if (prior == NULL)
                break;
            temp = prior->next;
            prior->next = NULL;
            reverseList(phead->next, prhead, prtail);
            phead->next = prhead;
            prtail->next = temp;
            prior = phead = prtail;
        }

        head = dummy->next;
        delete dummy;
        return head;
    }

    void reverseList(ListNode* head, ListNode* &prhead, ListNode* &prtail) {
        ListNode* p = head, *q = head->next, *temp;

        prtail = head;
        while (q)
        {
            temp = q->next;
            q->next = p;
            p = q;
            q = temp;
        }
        prhead = p;   
    }
};

 

上一篇:Learning Deep CNN Denoiser Prior for Image Restoration阅读笔记


下一篇:同一张表中,通过传入的id来获取所有关联的父id