题目来源:力扣
示例 1:
输入:root = [1,null,2,3] 输出:[1,2,3]
示例 2:
输入:root = [] 输出:[]
代码实现:
class Solution {
public:vector<int> preorderTraversal(TreeNode* root) {stack<TreeNode*> st;TreeNode* cur = root;vector<int> v; while(cur || !st.empty()){//访问一棵树的开始//1.访问左路节点,左路节点入栈while(cur){v.push_back(cur->val);st.push(cur);cur=cur->left;}//依次访问左路节点的右子树TreeNode* top = st.top();st.pop();//字问题的方式访问右子树cur = top->right;}return v;}
};
思路:
我们用这棵树来举例
我们在访问一棵树,按照前序遍历,我们最先访问的是左路节点,也就是8,3,1,所以我们就可以把树分为两个部分,一个是左路节点,另一个就是左路节点的右子树,我们把左路节点入栈
然后我们取栈顶元素,也就是1,我们访问完了1,然后pop一下,接着访问1的右子树,1的右子树为空,不用管
接着栈顶元素就是3了,我们访问3,然后pop一下,再访问3的右子树,也就是6,按照刚才的规则,把6和4入栈,接着就是4被访问,pop掉,6被访问,pop掉,访问6的右子树,也就是7,继续,7入栈,访问7,pop掉,然后7的左右为空,不用管
此时栈中就只剩8了,然后我们访问8,pop掉,再访问8的右子树,这个结果就是前序遍历了
我们要注意的是,cur指向一个节点代表访问一棵树的开始,栈里面的节点代表要依次访问它的右子树,大家可以画图理解一下