小米2019笔试 —— 打印二叉树_小米校招打印二叉树-优快云博客

本文介绍了一种从特定格式的字符串中解析并构建二叉树的方法。通过使用栈来匹配括号，确定子树的边界，实现了字符串到二叉树的转换。此方法适用于需要从文本描述中重建树结构的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

#include <iostream>
#include <vector>
#include <string>
#include <deque>
#include <stack>

using namespace std;

stack<char> s;

class Node{
public:
    Node* left;
    Node* right;
    int val;
    Node(int val):left(nullptr),right(nullptr),val(val){};
};

//begin指向当前节点
Node* build(string input, int begin, int end){
//    1(2(3,4(,5)),6(7,))
    //2(3,4(,5)),6(7,)
    //3,4(,5)    7,
    //4(,5)
    Node* head = NULL;
    if (begin <= end) {
        head = new Node(input[begin]-'0');
      	 
      	 //如果当前节点后有 ( 表示可以继续划分子树，否则就是叶子节点
        if (input[begin+1] == '(') {
            int cur = begin+2;
			
			//使用栈进行括号匹配，找到作为分隔符的那个 ,
            while (!s.empty() || input[cur]!=',') {
                if (input[cur] == '(') {
                    s.push('(');
                }else if (input[cur] == ')'){
                    s.pop();
                }
                cur++;
            }
            head->left = build(input, begin+2, cur-1);
            head->right = build(input, cur+1, end-1);
        }
    }
    return head;
}

void travel_mid(Node* head){
    if (head == NULL) {
        return;
    }
    travel_mid(head->left);
    cout << head->val;
    travel_mid(head->right);
}

int main(int argc, const char * argv[]) {
    cout << "start" << endl;
    string input;
    cin >> input;
    Node* head = build(input, 0, input.size()-1);
    travel_mid(head);
}