题目:
输入一个链表,按链表值从尾到头的顺序返回一个ArrayList。
这道题其实挺简单的,它的本质就是遍历链表,只是最后打印出来的顺序要倒序,因此可以直接采用栈,正序遍历时push进栈里,然后每次去除栈顶元素放入最终要返回的vector中并把它pop掉就行。代码如下:
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) :
* val(x), next(NULL) {
* }
* };
*/
class Solution {
public:
vector<int> printListFromTailToHead(ListNode* head) {
stack<int> num;
vector<int> res;
while (head != NULL) {
num.push(head->val);
head = head->next;
}
while (!num.empty()) {
res.push_back(num.top());
num.pop();
}
return res;
}
};
除了上面这种普通的解法以外,剑指还提供了另一种思路——递归。为什么联想到递归呢?因为递归在本质上就是一个栈结构,这点我之前还真的没有注意到。要实现倒序输出链表,只需要递归输出当前节点的后一个节点,最后再输出自己就可以了。结果发现我连递归都不会写了QAQ 所以在这里还是再重新理一下思路:递归基应该是最后一个节点,也就是当head->next为NULL时结束,这时候什么都不用做(底下的代码省略了这种情况),而当head->next不为NULL时,则要递归调用递归函数。那么递归函数里面的内容是什么呢,就是把当前节点的值push到vector中。举个栗子,当遍历到最后一个节点时,push进去了最后一个节点的val,在它的前一步是倒数第二个节点调用了print函数,调用完以后,倒数第二个节点的val也是需要被push进vector的,如此递归下去,就回到了第一个结点把它的val push进去。代码如下:
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) :
* val(x), next(NULL) {
* }
* };
*/
class Solution {
public:
vector<int> res;
vector<int> printListFromTailToHead(ListNode* head) {
if (head != NULL) {
if (head->next != NULL)
printListFromTailToHead(head->next);
res.push_back(head->val);
}
return res;
}
};
剑指上还指出,虽然递归版的代码比较简洁,但是当链表很长的时候,容易导致函数调用的层级很深,产生函数调用栈溢出的情况,因此鲁棒性还是没有循环实现的版本的好。