下面是C++实现非递归中序遍历二叉树的示例代码,使用栈来辅助实现遍历。首先,将输入的数据构造成二叉树,然后进行中序遍历。
- #include <iostream>
- #include <sstream>
- #include <vector>
- #include <stack>
- #include <string>
- using namespace std;
- // 定义二叉树节点结构体
- struct TreeNode {
- int val;
- TreeNode *left;
- TreeNode *right;
- TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
- };
- // 创建二叉树
- TreeNode* createTree(vector<string>& nodes) {
- if (nodes.empty() || nodes[0] == "None") return nullptr;
- TreeNode* root = new TreeNode(stoi(nodes[0]));
- vector<TreeNode*> nodeQueue = {root};
- int index = 1;
- // 通过字符串数组构建二叉树
- for (TreeNode* node : nodeQueue) {
- if (index < nodes.size() && nodes[index] != "None") {
- node->left = new TreeNode(stoi(nodes[index]));
- nodeQueue.push_back(node->left);
- }
- index++;
- if (index < nodes.size() && nodes[index] != "None") {
- node->right = new TreeNode(stoi(nodes[index]));
- nodeQueue.push_back(node->right);
- }
- index++;
- }
- return root;
- }
- // 二叉树的非递归中序遍历
- vector<int> inorderTraversal(TreeNode* root) {
- vector<int> result;
- stack<TreeNode*> s;
- TreeNode* currentNode = root;
-
- while (currentNode != nullptr || !s.empty()) {
- // 找到最左边的节点
- while (currentNode != nullptr) {
- s.push(currentNode);
- currentNode = currentNode->left;
- }
-
- // 访问栈顶节点
- currentNode = s.top();
- s.pop();
- result.push_back(currentNode->val);
-
- // 访问右子树
- currentNode = currentNode->right;
- }
-
- return result;
- }
- int main() {
- string input;
- getline(cin, input);
- istringstream iss(input);
- vector<string> nodes;
- string node;
- while (iss >> node) {
- nodes.push_back(node);
- }
-
- TreeNode* root = createTree(nodes);
- vector<int> result = inorderTraversal(root);
-
- for (size_t i = 0; i < result.size(); ++i) {
- if (i > 0) cout << " ";
- cout << result[i];
- }
- cout << endl;
-
- return 0;
- }
复制代码
**代码说明:**
- 首先定义了一个二叉树的节点结构。
- 使用`createTree`函数根据输入创建二叉树。
- 使用`inorderTraversal`函数进行非递归中序遍历。
- 主函数中读取输入,调用以上函数,最后输出结果。
你可以根据这个思路用C实现。C++标准库中提供的`stack`类是实现非递归遍历的关键部分。
以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。