洛谷P1087 [NOIP2004 普及组] FBI 树（树，后序遍历，模拟）

最新推荐文章于 2023-08-01 19:20:34 发布

b_b_lai_lai

最新推荐文章于 2023-08-01 19:20:34 发布

阅读量164

点赞数

文章标签： c++ 算法

本文链接：https://blog.csdn.net/b_b_lai_lai/article/details/114548668

版权

这篇博客主要介绍了一个使用C++实现的递归算法，用于根据输入的字符数组构建一棵二叉树，并根据节点的0/1状态输出相应的字符（B/I/F）。博主分享了代码调试过程中的困扰以及解决方案，展示了从全0子树到全1子树再到混合子树的判断逻辑。

摘要由CSDN通过智能技术生成

题解：暴力模拟过的，debug了好久，结果一看题解我哭了

#include <bits/stdc++.h>
using namespace std;
inline int read() {
    int x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9')
        x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
    return x * f;
}

int n, sum = 1;
struct node {
    int l, r;
    int val;
} vis[100000];
char p1[1050];
int cnt, p[1050];
inline int tree(int l, int r) {
    if (l > r) {
        return 0;
    }
    cnt++;
    if (l == r) {
        vis[cnt].val = p[l] + 1;
        return cnt;
    }
    int now = cnt;
    int mid = (l + r) / 2;
    vis[now].l = tree(l, mid);
    vis[now].r = tree(mid + 1, r);
    if (vis[now].l && vis[now].r) {
        if (vis[vis[now].l].val == 1 && vis[vis[now].r].val == 1)
            vis[now].val = 1;  //全0
        else if (vis[vis[now].l].val == 2 && vis[vis[now].r].val == 2)
            vis[now].val = 2;  //全1
        else if (vis[vis[now].l].val == 3 && vis[vis[now].r].val == 3)
            vis[now].val = 3;  //有0有1
        else if (vis[vis[now].l].val + vis[vis[now].r].val >= 3)
            vis[now].val = 3;  //有0有1
    } else {
        if (vis[vis[now].l].val + vis[vis[now].r].val == 1)
            vis[now].val = 1;  //全0
        else if (vis[vis[now].l].val + vis[vis[now].r].val == 2)
            vis[now].val = 2;  //全1
        else if (vis[vis[now].l].val + vis[vis[now].r].val == 3)
            vis[now].val = 3;  //有0有1
    }
    return now;
}
inline void pintf(int x) {
    if (vis[x].l) {
        pintf(vis[x].l);
    }
    if (vis[x].r) {
        pintf(vis[x].r);
    }
    if (vis[x].val == 1) {
        cout << "B";
    } else if (vis[x].val == 2) {
        cout << "I";
    } else if (vis[x].val == 3) {
        cout << "F";
    }
}
int main() {
    n = read();
    for (int i = 0; i < n; ++i) {
        sum *= 2;
    }
    cin >> p1;
    for (int i = 0; i < sum; ++i) {
        p[i] = p1[i] - '0';
    }
    tree(0, sum - 1);
    pintf(1);
    return 0;
}