剑指 Offer 06. 从尾到头打印链表
一、题目:
输入一个链表的头节点,从尾到头反过来返回每个节点的值(用数组返回)。
示例 1:
输入:head = [1,3,2]
输出:[2,3,1]
限制:
0 <= 链表长度 <= 10000
二、思路代码和结果:
方法一:
题目中说从尾到头返回,那么在我的记忆中有一种从头而降,不是,有一种先进后厨的数组结构->栈!
具体做法:
把链表逐个放到栈中,然后把栈顶依次放到数组中(题目要求用数组返回),这时候就已经是逆序了,满足题意。
代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
vector<int> reversePrint(ListNode* head) {
stack<int>st;//定义栈
vector<int>res;//定义数组
while(head != NULL){//链表元素依次入栈
st.push(head->val);
head=head->next;
}
while(st.size()){//当栈不为空的时候
res.push_back(st.top());//栈顶放到数组中去
st.pop();//栈顶出栈
}
return res;
}
};
结果:
方法二:
有了一以后,可以小小修改一下:
题目要求用数组返回,如果用数组返回的话我们只需将链表元素逐个放到数组中去,然后将数组逆向输出即可。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
vector<int> reversePrint(ListNode* head) {
vector<int>res;//定义数组
while(head != NULL){//链表元素依次入数组
res.push_back(head->val);
head=head->next;
}
reverse(res.begin(),res.end());//stl reverse 函数 将数组翻转
return res;
}
};
对比:用栈似乎更快,用空间换时间;用数组翻转的话占用更少的空间。
怕什么真理无穷,进一步有进一步的欢喜!