输入一个链表的头节点,从尾到头反过来返回每个节点的值(用数组返回)。
示例 1:
输入:head = [1,3,2]
输出:[2,3,1]
解题思路(辅助栈法):把链表翻转过来,很显然可以把链表的数据域存入栈中,利用栈的后进先出的特点,再存入数组就实现了翻转。
算法流程:
1.创建辅助栈st和数组arr
2.用循环把链表的数据域存入栈st中
3.用循环把栈中的元素存入数组arr中
4.返回数组arr
代码:
class Solution {
public:
vector<int> reversePrint(ListNode* head) {
// 第一步
stack<int>st;
vector<int>arr;
// 第二步
while(head!=nullptr) {
st.push(head->val);
head=head->next;
}
// 第三步
while(!st.empty()) {
arr.push_back(st.top());
st.pop();
}
第四步
return arr;
}
};
解题思路(递归法):想到栈就想到递归,因为递归也是一种后进先出。调用递归到链表的最后一个结点,然后把该结点的数据域赋给数组arr,再返回arr,一直返回到第一个元素。
算法流程:
1.定义一个数组arr(不在该递归函数内定义)
2.使用递归直到最后一个元素,还需判断链表是否为空链表,是空链表直接返回数组arr
3.返回数组arr
代码:
class Solution {
public:
// 第一步
vector<int>arr;
// 递归函数
vector<int> reversePrint(ListNode* head) {
// 第二步
if(head!=nullptr) {
if(head->next!=nullptr) {
reversePrint(head->next);
}
arr.push_back(head->val);
}
// 第三步
return arr;
}
};