SDUTOJ2824求二叉树的层次遍历

版权声明：iQXQZX https://blog.csdn.net/Cherishlife_/article/details/84986154

求二叉树的层次遍历

https://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/contestproblem/cid/2711/pid/2824

Time Limit: 1000 ms Memory Limit: 65536 KiB

Submit Statistic

Problem Description

已知一颗二叉树的前序遍历和中序遍历，求二叉树的层次遍历。

Input

输入数据有多组，输入T，代表有T组测试数据。每组数据有两个长度小于50的字符串，第一个字符串为前序遍历，第二个为中序遍历。

Output

每组输出这颗二叉树的层次遍历。

Sample Inpu

2
abc
bac
abdec
dbeac

Sample Output

abc
abcde

Hint

Source

fmh

AC代码：

#include <bits/stdc++.h>
using namespace std;
struct tree
{
    char c;
    tree *l, *r;
};
char s1[100], s2[100]; // s1前序 s2中序
tree *make(int len, char *s1, char *s2) /* 根据前序中序建立二叉树*/
{
    int i;
    tree *root;
    if (len <= 0) // 递归边界 如果len为0则不能建树
        return NULL;
    root = new tree; // new一个新节点
    root->c = s1[0];
    for (i = 0; i < len; i++)
    {
        if (s2[i] == s1[0]) // 在中序中寻找根节点
            break;
    }
    root->l = make(i, s1 + 1, s2); // 递归建立左子树
    root->r = make(len - i - 1, s1 + i + 1, s2 + i + 1); // 递归建立右子树
    return root;
}
void show(tree *root) // 层序遍历二叉树 // 通过队列实现
{
    queue<tree *> q;
    if (root) // 树存在 入队  手动模拟一下即可明白题意
        q.push(root);
    while (!q.empty())
    {
        root = q.front();
        q.pop();
        if (root)
        {
            cout << root->c;
            q.push(root->l);
            q.push(root->r);
        }
    }
}
int main()
{
    int n;
    cin >> n;
    while (n--)
    {
        tree *tree;
        cin >> s1 >> s2;
        int len = strlen(s1);
        tree = make(len, s1, s2);
        show(tree);
        cout << endl;
    }
    return 0;
}