题目:https://leetcode-cn.com/problems/binary-tree-preorder-traversal/submissions/
思路:使用迭代方式 前序遍历是中左右,每次先处理的是根节点,那么先将根节点放入栈中,在将右孩子 放入进去,最后放入做孩子,(栈是先进后出),这样才能做到西安处理左孩子,在处理右孩子
class Solution { public: vector preorderTraversal(TreeNode* root) { vectorvec; stack<TreeNode*>stk; stk.push (root); while(!stk.empty()) { TreeNode*node=stk.top(); stk.pop(); if(node!=NULL) vec.push_back(node->val); //先对根节点进行处理 else continue; stk.push(node->right); //放入右孩子 stk.push(node->left); //放入左孩子 } return vec;
}
};
思路二:使用递归方式
class Solution { public: vector res; vector preorderTraversal(TreeNode* root) { if(root) { res.push_back(root->val); //根 preorderTraversal(root->left); //左 preorderTraversal(root->right); //右 } return res; } };
在写递归时犯了一个错误,我将 vector res;这一句写在了函数里面 输出出错,这是因为 每递归一次时就会重新执行这一句,当递归结束在栈中创建的局部变量就会消失,(递归的数据结构就是栈)这样将会只保留开始入栈的那个数,因此指挥输出一个根节点