資料結構實驗之二叉樹八:(中序後序)求二叉樹的深度

鵬不是這個朋發表於2020-10-30

Description

已知一顆二叉樹的中序遍歷序列和後序遍歷序列,求二叉樹的深度。
Input

輸入資料有多組,輸入T,代表有T組資料。每組資料包括兩個長度小於50的字串,第一個字串表示二叉樹的中序遍歷,第二個表示二叉樹的後序遍歷。
Output

輸出二叉樹的深度。
Sample
Input

2

dbgeafc

dgebfca

lnixu

linux

Output

4

3

#include<bits/stdc++.h>

using namespace std;

typedef struct node
{
    char data;
    struct node *l, *r;
} Tree;

char mid[55], pos[55];

Tree *creat(char *mid, char *pos, int len)
{
    Tree *root;
    if(len == 0)
        return NULL;
    root = new Tree;
    root->data = pos[len - 1];
    int i;
    for(i = 0; i < len; i++)
    {
        if(mid[i] == pos[len - 1])
            break;
    }
    root->l = creat(mid, pos, i);
    root->r = creat(mid + i + 1, pos + i, len - i - 1);
    return root;
}

int depth_bintree(Tree *root)
{
    int de = 0;
    if(root)
    {
        int left_depth = depth_bintree(root->l);
        int right_depth = depth_bintree(root->r);
        de = left_depth > right_depth ? left_depth + 1 : right_depth + 1;
    }
    return de;
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%s", mid);
        scanf("%s", pos);
        int len = strlen(pos);
        Tree *root = creat(mid, pos, len);
        int depth= depth_bintree(root);
        printf("%d\n", depth);
    }
    return 0;
}

相關文章