Leetcode K个一组反转链表C++实现

最新推荐文章于 2024-09-01 16:39:39 发布

原创最新推荐文章于 2024-09-01 16:39:39 发布 · 702 阅读

3 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用栈解决LeetCode中K个一组翻转链表问题的方法。通过遍历链表并将节点存入栈，当栈内节点数达到K时，弹出节点并逆序构建新链表。对于剩余不足K个的节点，同样进行逆序处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目要求：

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

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

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

基本思路：

采用了栈来完成，遍历整个链表，并将结点逐个存入栈中，当栈中存放结点数量=k时，将栈中元素依次弹出，创建一个新的链表。值得注意的是，当最后剩下的数不足k个时，此时虽然剩下结点全都在栈中，但还是要将其逆序，弹入另外一个栈中，然后再弹出。

Answer：

/**
 * 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||head->next==NULL||k==1) return head;
        ListNode* cur = head;
        int i = 1;
        while(cur->next!=NULL){
            i++;
            cur =cur->next;
        }
        if(i<k) return head;
        stack<ListNode*> s;
        ListNode* p = head;
        while(s.size()!=k){
            s.push(p);
            p = p->next;
        }
        ListNode* phead = s.top();
        s.pop();
        ListNode* p2 = phead;
        while(!s.empty()){
            p2->next = s.top();
            s.pop();
            p2 = p2->next;
            p2->next = NULL;
        }
        while(p!=NULL){
            s.push(p);
            p = p->next;
            if(s.size()==k){
                while(!s.empty()){
                    p2->next = s.top();
                    s.pop();
                    p2 = p2->next;
                    p2->next = NULL;
                }
            }
        }
        if(!s.empty()){
            stack<ListNode*> s2;
            while(!s.empty()){
                s2.push(s.top());
                s.pop();
            }
            while(!s2.empty()){
                p2->next = s2.top();
                s2.pop();
                p2 = p2->next;
            }
        }
        return phead;
    }
};