一、题目
以上面二叉树为例,前序遍历为:ABDECFG。
怎么实现这个顺序呢 ?
二、递归的实现
按照访问根节点——左子树——右子树的方式遍历这棵树,而在访问左子树或者右子树的时候,我们按照同样的方式遍历,直到遍历完整棵树。因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。
class Solution {
private:
vector<int> res;
public:
vector<int> preorderTraversal(TreeNode* root) {
if(root == nullptr) {
return {};
}
res.push_back(root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
return res;
}
};
三、迭代的实现
前序遍历二叉树是先访问根结点,然后遍历左子树,再接着是遍历右子树。
首先我们应该创建一个Stack用来存放节点,首先我们想要打印根节点的数据,此时Stack里面的内容为空,所以我们优先将头结点加入Stack,然后打印。
之后我们应该先打印左子树,然后右子树。所以先加入Stack的就是右子树,然后左子树。
此时你能得到的流程如下:
class Solution {
private:
vector<int> res;
public:
vector<int> preorderTraversal(TreeNode* root) {
if(root == nullptr) {
return {};
}
stack<TreeNode*> stack;
stack.push(root);
while (!stack.empty()){
TreeNode* topNode = stack.top();
stack.pop();
res.push_back(topNode->val);
if (topNode->right) {
stack.push(topNode->right);
}
if (topNode->left) {
stack.push(topNode->left);
}
}
return res;
}
};
另一种写法:
根据前序遍历的顺序,优先访问根结点,然后在访问左子树和右子树。所以,对于任意结点node,第一部分即直接访问之,之后在判断左子树是否为空,不为空时即重复上面的步骤,直到其为空。若为空,则需要访问右子树。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if (root == nullptr) {
return res;
}
stack<TreeNode*> stk;
TreeNode* node = root;
while (!stk.empty() || node != nullptr) {
while (node != nullptr) {
res.emplace_back(node->val);
stk.emplace(node);
node = node->left;
}
node = stk.top();
stk.pop();
node = node->right;
}
return res;
}
};
或者
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if (root == nullptr) {
return res;
}
stack<TreeNode*> stk;
TreeNode* node = root;
while (!stk.empty() || node != nullptr) {
if (node != nullptr) {
res.emplace_back(node->val);
stk.emplace(node);
node = node->left;
}
else {
node = stk.top();
stk.pop();
node = node->right;
}
}
return res;
}
};
参考: