文章目录
题目描述
解题思路
递归方法处理,利用栈的后进先出的特性,结合栈+递归来处理本题的反转链表。
构造一个容积为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;
}
};