UVA122 trees on the level(BFS)

最新推荐文章于 2022-08-31 11:06:24 发布

原创最新推荐文章于 2022-08-31 11:06:24 发布 · 450 阅读

1 ·

CC 4.0 BY-SA版权

UVA 专栏收录该内容

232 篇文章

订阅专栏

本文详细阐述了一个自定义二叉树构建算法，并通过宽度优先搜索（BFS）验证其正确性。重点包括输入读取、树构建、内存管理以及输出展示流程。

写了n遍总算是写对了，不能停止地相信自己。

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 300;
struct Node
{
    bool have_value;//标记是否赋值
    int v;//值
    Node *left, *right;//左节点和右节点
    Node():have_value(false),left(NULL),right(NULL){}//成员函数，初始化结构体变量
};
Node* root = NULL;//根节点root
char s[maxn];//存输入的字符数组
bool failed;//标记正确错误的变量
Node* newnode()//把new Node()封装到newnode()里
{
    return new Node();
}
void addnode(int v, char * s)//创建链表，形成二叉树
{
    int n = strlen(s);//s的长度
    Node* u = root;//把根节点赋给u
    for(int i = 0; i < n; i++)//便利LR字符串
    {
        if(s[i] == 'L'){
            if(u->left == NULL) u->left = newnode();//左节点为空，创建新节点
            u = u->left;//指向左节点
        }else if(s[i] == 'R'){
            if(u->right == NULL) u->right = newnode();//右节点为空，创建新节点
            u = u->right;//指向右节点
        }
    }
    if(u->have_value) failed = true;//如果已经有值了，出项了错误
    u->v = v;//v赋给当前节点的v
    u->have_value = true;//复制成功，标记为已有值
}
void remove_tree(Node* u)//释放内存
{
    if(u == NULL) return;//当为0时 停止释放
    remove_tree(u->left);//先释放左节点
    remove_tree(u->right);//再释放右节点
    delete u;//删除u
}

bool read_input()//读入二叉树
{
    failed = false;//开始标记failed为false
    remove_tree(root);//释放上一个二叉树的内存
    root = newnode();//初始化root的值
    for(;;)//读取本次二叉树所用的循环
    {
        if(scanf("%s", s) != 1) return false;//EOF输入结束
        if(!strcmp(s, "()")) break;//出现()本次输入结束
        int v;//定义值
        sscanf(&s[1],"%d", &v);//从字符中读取数值

        addnode(v, strchr(s, ',')+1);//创建链表，形成二叉树
    }
    return true;
}
bool bfs(vector<int>& ans)//宽度优先队列
{
    queue<Node*> q;//定义Node*类型的队列
    ans.clear();//清空不定长数组ans
    q.push(root);//根节点（根节点本来就是指向结构体的指针）进栈
    while(!q.empty())//只要q不为空，便利所有不为空的节点
    {
        Node* u = q.front();//u为队首结构体指针
        q.pop();//赋值后出栈
        if(!u->have_value) return false;//如果他没有值，错误了
        ans.push_back(u->v);//想不定长数组尾添加当前结构体的v
        if(u->left != NULL) q.push(u->left);//先向左便利，后向右便利
        if(u->right != NULL) q.push(u->right);
    }
    return true;
}

int main()
{
    vector<int> ans;//定义的不定长数组，存入是每个二叉树的值
    while(read_input())//读入，返回FALSE时结束
    {
        if(!bfs(ans)) failed = true;//当bfs返回为false，出现错误
        if(failed) printf("not complete\n");
        else {//输出每个结构体的值
            for(int i= 0 ; i < ans.size(); i++)
            {
                if(i != 0)  printf(" ");
                printf("%d", ans[i]);
            }
            printf("\n");
        }
    }
    return 0;
}

注释都为自己写的，如有纰漏，请大神指正。