剑指Offer 06. 从尾到头打印链表
每日几道leetcode刷刷题!
传送门
题目描述
输入一个链表的头节点,从尾到头反过来返回每个节点的值(用数组返回)。
示例 1:
输入:head = [1,3,2]
输出:[2,3,1]
代码
Python版
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def reversePrint(self, head: ListNode) -> List[int]:
stack = []
while head:
stack.append(head.val)
head = head.next
return stack[::-1]
C++版
/**
* 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;
stack<int> s;
while (head){
s.push(head->val);
head = head->next;
}
while (!s.empty()){
res.push_back(s.top());
s.pop();
}
return res;
}
};
总结
- 反转一个数组,a[::-1]
- stack是标准库中的一个容器适配器,是个类模板,使用的时候需要实例化,int是模板实参。
- stack<int> st声明了1个存储int型元素的栈,栈名是
st
。 - C++中
s.empty()
作为判断容器是否为空的函数 - push_back是编程语言里面的一个函数名。如c++中的vector头文件里面就有这个push_back函数,在vector类中作用为在vector尾部加入一个数据。
- C ++ Stack
top()
函数的作用是:返回栈顶元素的值。(只是返回,但不弹出) - pop()弹出栈顶