加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
DFS.cpp 1.95 KB
一键复制 编辑 原始数据 按行查看 历史
yuetan 提交于 2020-09-21 00:02 . 二叉树的遍历
#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <algorithm>
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
/* 迭代法 前(根)序遍历 */
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> stack0;
TreeNode* cur = root;
while (cur || stack0.size()) {
while (cur) {
stack0.push(cur);
res.push_back(cur->val);
cur = cur->left;
}
cur = stack0.top();
stack0.pop();
cur = cur->right;
}
return res;
}
/* 迭代法 中(根)序遍历 */
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> stack0;
TreeNode* cur = root;
while (cur || stack0.size()) {
while (cur) {
stack0.push(cur);
cur = cur->left;
}
cur = stack0.top();
stack0.pop();
res.push_back(cur->val);
cur = cur->right;
}
return res;
}
/* 迭代法 后(根)序遍历 */
/* 后序遍历是参考前序遍历 利用“根-右-左”的顺序代替“根-左-右”的顺序,然后进行vector转置得到*/
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> stack0;
TreeNode* cur = root;
while (cur || stack0.size()) {
while (cur) {
stack0.push(cur);
res.push_back(cur->val);
cur = cur->right;
}
cur = stack0.top();
stack0.pop();
cur = cur->left;
}
reverse(res.begin(), res.end()); //头文件algorithm
return res;
}
};
int main()
{
Solution sol;
return 0;
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化