top5-k个一组反转链表

题目描述

解题思路

递归方法处理，利用栈的后进先出的特性，结合栈+递归来处理本题的反转链表。
构造一个容积为k - 1的栈，当达到体积后判断是否达到终止条件即：cur->next是否为nullptr，如果为则说明到达结尾开始回溯。如果没有到达则继续递归处理。（cur->next,k）。到达终止条件后开始处理栈里面的数据。我们将当前节点指向head的next指针。
然后创建一个临时节点存储当前节点，我们该次递归状态下的栈中数据，将当前节点的next节点指向栈中的节点（而栈中节点保存的是上一次的进栈数据因此完成了反转）以此递归下去，最终返回最外层递归的cur即可。

#include<iostream>
#include<vector>
#include<sstream>
#include<stack>
using namespace std;
vector<int> op_data;
string input;

struct ListNode
{
    int val;
    ListNode* next;
    ListNode():val(0), next(nullptr){}
    ListNode(int _val):val(_val), next(nullptr){}
    ListNode(int _val, ListNode* _myList):val(_val), next(_myList){}
};

ListNode* reverseKGroup(ListNode* head, int k) 
{
    int t = k;
    ListNode* cur = head;
    stack<ListNode*> stk;

    while(-- t && cur != nullptr)
    {
        stk.push(cur);
        cur = cur->next;
    }
    
    //若节点个数不足k个
    if(cur == nullptr) return head;
    
    //递归处理子节点的问题
    //*****************************************************************************************************
    head->next = reverseKGroup(cur->next, k);
    ListNode* tmp = cur;
    while(stk.size())
    {
        tmp->next = stk.top();
        stk.pop();
        tmp = tmp->next;
    }
    return cur;
}

int main()
{
    ListNode* node0 = new ListNode(1);
    ListNode* node1 = new ListNode(2);
    ListNode* node2 = new ListNode(3);
    ListNode* node3 = new ListNode(4);
    ListNode* node4 = new ListNode(5); 
    ListNode* node5 = new ListNode(6); 
    ListNode* node6 = new ListNode(7); 
    
    node0->next = node1;
    node1->next = node2;
    node2->next = node3;
    node3->next = node4;
    node4->next = node5;
    node5->next = node6;
    node6->next = nullptr;
    ListNode* res = reverseKGroup(node0, 3);
    while(res != nullptr) 
    {
        cout << res->val << "->";
        res = res->next;
    }
    
    
    return 0;
}

Leetcode模式

class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        int t = k;
        ListNode* cur = head;
        stack<ListNode*> stk;
        while(-- t && cur != nullptr)
        {
            stk.push(cur);
            cur = cur-> next;
        }
        if(cur == nullptr) return head;

        head->next = reverseKGroup(cur->next, k);
        ListNode * tmp = cur;
        while(stk.size())
        {
            tmp->next = stk.top();
            stk.pop();
            tmp = tmp->next;
        }
        return cur;
    }
};