【剑指Offer-06】从尾到头打印链表

问题

输入一个链表的头节点，从尾到头反过来返回每个节点的值（用数组返回）。

 // Definition for singly-linked list.
 struct ListNode {
     int val;
     ListNode *next;
     ListNode(int x) : val(x), next(NULL) {}
 };

示例

输入：head = [1,3,2]
输出：[2,3,1]

解答1：栈

class Solution {
public:
    vector<int> reversePrint(ListNode* head) {
        if (!head) return {};
        stack<int> s;
        vector<int> res;
        ListNode* cur = head;
        while (cur) {
            s.push(cur->val);
            cur = cur->next;
        }
        while (!s.empty()) {
            res.push_back(s.top());
            s.pop();
        }
        return res;
    }
};

重点思路

遍历链表，将对应值依次压入栈。再将栈中数据依次pop到结果数组中。

解答2：递归

class Solution {
public:
    vector<int> reversePrint(ListNode* head) {
        if (!head) return {};
        vector<int> res = reversePrint(head->next);
        res.push_back(head->val);
        return res;
    }
};

重点思路

栈可以压入数字结果，同样也可以压入函数，后者即为递归做法。使用递归代码会更加简洁，但是要注意当链表很长时可能导致函数调用栈溢出。