题目来源:力扣
示例 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指向一个节点代表访问一棵树的开始,栈里面的节点代表要依次访问它的右子树,大家可以画图理解一下