面试题06. 从尾到头打印链表

1.题目

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

示例 1：

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

2. 思路

2.1递归

1. 递推阶段： 每次传入 head.next ，以 head == null（即走过链表尾部节点）为递归终止条件，此时直接返回。
2. 回溯阶段：层层回溯时，将当前节点值加入列表，即tmp.add(head.val)。
3. 最终，将列表 tmp 转化为数组 res ，并返回即可。

• 时间复杂度 O(N)： 遍历链表，递归 N 次。
• 空间复杂度 O(N)： 系统递归需要使用 O(N)的栈空间。

class Solution {
    ArrayList<Integer> tmp = new ArrayList<Integer>();
    public int[] reversePrint(ListNode head) {
        recur(head);
        int[] res = new int[tmp.size()];
        for(int i = 0; i < res.length; i++)
            res[i] = tmp.get(i);
        return res;
    }
    void recur(ListNode head) {
        if(head == null) return;
        recur(head.next);
        tmp.add(head.val);
    }
}

2.2 辅助栈

1. 入栈： 遍历链表，将各节点值 push 入栈。（借助 LinkedList 的addLast()方法）。
2. 出栈： 将各节点值 pop 出栈，存储于数组并返回。（​新建一个数组，通过 popLast() 方法将各元素存入数组，实现倒序输出）。

• 时间复杂度 O(N)： 入栈和出栈共使用 O(N) 时间。
• 空间复杂度 O(N)： 辅助栈 stack 和数组 res 共使用 O(N)的额外空间。

class Solution {
    public int[] reversePrint(ListNode head) {
        LinkedList<Integer> stack = new LinkedList<Integer>();
        while(head != null) {
            stack.addLast(head.val);
            head = head.next;
        }
        int[] res = new int[stack.size()];
        for(int i = 0; i < res.length; i++)
            res[i] = stack.removeLast();
    return res;
    }
}