剑指Offer06: 从尾到头打印链表
题目描述:
输入一个链表的头节点,从尾到头反过来返回每个节点的值(用数组返回)
示例:
输入:head = [1,3,2]
输出:[2,3,1]
解题思路1:
使用栈先进后出的特性,构造一个辅助栈用来将链表的元素逆置。
对栈进行稍微的复习
栈是Vector的一个子类,它实现了一个标准的后进先出的栈。
堆栈只定义了默认构造函数,用来创建一个空栈。 堆栈除了包括由Vector定义的所有方法,也定义了自己的一些方法。
- boolean empty() 栈是否为空
- Object peek() 查看栈顶元素
- Object pop() 弹出栈顶元素
- Object push(Object element) 将元素压入栈
- int search(Object element) 返回元素在栈中的位置,以1为基数
具体的步骤如下:
- 正向遍历一遍链表,构造栈
- 将栈中的元素逐个弹出,形成一个数组
时间复杂度和空间复杂度都是O(n)
代码如下:
public int[] reversePrint(ListNode head) {
//使用栈
Stack<ListNode> stack = new Stack<ListNode>();
ListNode temp =head;
while(temp != null){
stack.push(temp);
temp=temp.next;
}
int size = stack.size();
int[] print = new int[size];
for (int i = 0; i < size; i++) {
print[i]=stack.pop().val;
}
return print;
//时间和空间复杂度都是O(n)
//2 ms 38.9MB
}
同种思路下用linkedList实现如下
linkedList可以看成双端链表,双端链表是可以实现栈结构,双端链表的插入和删除要比栈要快一点。
粗略地来说:Stack是用数组来实现的,进栈和出栈需要对后面的元素进行调整。而双端链表不需要
public int[] reversePrint1(ListNode head){
LinkedList<Integer> stack = new LinkedList<>();
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;
//1ms 39.1MB
}
解题思路2:
使用递归的思想
先走至链表末端,回溯时依次将节点值加入列表,这样就可以实现链表值的倒序输出。
算法流程:
- 递推阶段:每次传入
head.next
,以head == null
(即走过链表尾部节点)为递归终止条件,此时直接返回。 - 回溯阶段: 层层回溯时,将当前节点值加入列表,即
tmp.add(head.val)
。 - 最终,将列表
tmp
转化为数组res
,并返回即可。
复杂度分析:
时间复杂度和空间复杂度都是O(N)
从程序提交的效果上来看,递归要比双端链表实现的辅助栈内存空间要大一点,这也在意料之中,除了构建一个辅助的动态数组之外,还离不开函数栈。在进行递归操作时,不断地把之前压入函数栈的信息弹出。
补充ArrayList相关知识
ArrayList继承AbstractList,实现List接口
常用方法:
- add()添加元素
- remove()删除元素
- size()计算元素数量
完整代码如下:
//使用递归
ArrayList<Integer> tmp = new ArrayList<Integer>();
public int[] reversePrint3(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);
}
//1 ms 40.4 MB