7-31 二叉树查找结点及父结点

编写程序在二叉树中查找给定结点及父结点。二叉树结点的数据域值不等于0的整数。

输入格式:

输入第1行为一组用空格间隔的整数，表示带空指针信息的二叉树先根序列，其中空指针用0表示。例如1 5 8 0 0 0 6 0 0表示如下图的二叉树。第2行为整数m，表示查询个数。接下来m行，每行为一个不等于0的整数K，表示要查找的结点的数据值。m不超过100，二叉树结点个数不超过150000，高度不超过6000。输入数据保证二叉树各结点数据值互不相等。

输出格式:

输出为m行，每行1个整数，表示被查找结点K的父结点数据值，若二叉树中无结点K或结点K无父结点，则输出0。

输入样例:

1 5 8 0 0 0 6 0 0
3
8
1
6

输出样例:

5
0
1

代码长度限制

16 KB

时间限制

300 ms

内存限制

20 MB

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>
typedef struct TNode* Position;
typedef Position BinTree;
typedef struct TNode
{
    int Data;
    BinTree Left;
    BinTree Right;
    BinTree Father;
};
int a[15000];
int k = 0;
int flag;
void CreateTree(BinTree* BT)
{
    int val;
    scanf("%d", &val);
    if (val == 0)
        *BT = NULL;
    else
    {
        *BT = (BinTree)malloc(sizeof(struct TNode));
        (*BT)->Data = val;
        (*BT)->Father = NULL;
        CreateTree(&(*BT)->Left);
        CreateTree(&(*BT)->Right);
    }
}

void CreateFatherNode(BinTree BT)
{
    if (BT->Left != NULL)
    {
        BT->Left->Father = BT;
        CreateFatherNode(BT->Left);

    }if (BT->Right != NULL)
    {
        BT->Right->Father = BT;
        CreateFatherNode(BT->Right);
    }
}

int  PreorderTraversal(BinTree BT, int n)
{
    int p;
    if (BT->Data == n)
    {
        flag = 1;
        if (BT->Father == NULL)
        {
            a[k] = 0;
            k++;
        }
        else
        {
            p = (BT->Father->Data);
            a[k] = p;
            k++;
        }
        return flag;
    }
    if (BT->Left != NULL && flag == 0)
    {
        PreorderTraversal(BT->Left, n);
    }
    if (BT->Right != NULL && flag == 0)
        PreorderTraversal(BT->Right, n);
}

int main()
{
    BinTree BT;
    int m, i, j, n;
    BT = (BinTree)malloc(sizeof(struct TNode));
    CreateTree(&BT);
    CreateFatherNode(BT);
    scanf("%d", &m);
    if (m <= 100)
    {
        for (i = 0; i < m; i++)
        {
            flag = 0;
            scanf("%d", &n);
            PreorderTraversal(BT, n);
            if (flag == 0)
            {
                a[k] = 0;
                k++;
            }
        }
        for (j = 0; j < k; j++)
            printf("%d\n", a[j]);
    }
    return 0;
}