/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode *>st;
vector<int> res;
if(root == NULL) return res;
st.push(root);
//如果栈为空结束循环
while(!st.empty()){
//取出栈顶元素
TreeNode *node = st.top();
st.pop();//弹出
//如果取出的元素不为空,则加入到数组中,否则跳过(因为该节点已经空了)
if(node != NULL){
res.push_back(node->val);
}
else{
continue;
}
st.push(node->right);
st.push(node->left);
}
return res;
}
};
思路:创建栈来处理根节点的元素,当根节点不为空时,将根节点压入栈中,由于栈的性质(先进后出),所以先将节点的右节点压入栈中,然后再压入左节点。
此时处理节点,先把中节点放入栈中,判断栈是否为空,不为空则将栈顶的元素弹出,判断元素是否为空,不为空则加入数组,然后将右节点跟左节点压入栈中。如果为空,则不将元素加入数组。