题目描述
算法1
递归
step 1:准备数组用来记录遍历到的节点值,Java可以用List,C++可以直接用vector。
step 2:从根节点开始进入递归,遇到空节点就返回,否则将该节点值加入数组。
step 3:依次进入左右子树进行递归。
C++ 代码
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型vector
*/
void preorder(vector<int> &res, TreeNode* root){
if (root == NULL)
return;
res.push_back(root->val);
preorder(res,root->left);
preorder(res,root->right);
}
vector<int> preorderTraversal(TreeNode* root) {
// write code here
vector<int> res;
preorder(res,root);
return res;
}
};
算法2
栈
C++ 代码
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if(root == NULL)
return res;
//辅助栈
stack<TreeNode*> s;
//根节点先进栈
s.push(root);
//直到栈中没有节点
while(!s.empty()){
//每次栈顶就是访问的元素
TreeNode* node = s.top();
s.pop();
res.push_back(node->val);
//如果右边还有右子节点进栈
if(node->right)
s.push(node->right);
//如果左边还有左子节点进栈
if(node->left)
s.push(node->left);
}
return res;
}
};