判断一棵二叉树是否为搜索二叉树和完全二叉树

判断一棵二叉树是否为搜索二叉树和完全二叉树

题目描述

给定一棵二叉树，已经其中没有重复值的节点，请判断该二叉树是否为搜索二叉树和完全二叉树。

输入描述：

第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。

以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)

ps:节点的标号就是节点的值

输出描述：

第一行输出该二叉树是否为搜索二叉树的答案，如果是则输出 “true”，否则输出 “false”。

第二行输出该二叉树是否为完全二叉树的答案，如果是则输出 “true”，否则输出 “false”。

备注：

$1\le n\le 5\ast 10^5$

$1\le fa,lch,rch,root\le n$

---

题解：

判断二叉搜索树：递归进行中序遍历就行了，二叉搜索树的中序遍历一定是升序的，可以保存中序遍历的前一个节点和当前节点进行判断，一旦不满足直接退出

判断完全二叉树：按照以下规则进行：

• 层次遍历，从左到右的顺序；
• 如果当前节点有 右儿子 ，没有 左儿子 ，返回 false；
• 如果当前节点并不是 左右儿子 都有，那么后面的节点必须都为 叶节点，否则返回 false；
• 返回 true

代码：

#include <cstdio>
#include <queue>

using namespace std;

const int N = 500010;

struct BST {
    int lch, rch;
} bst[N];

int n, rt;
int fa, lch, rch;

bool is_bst(int root, int& pre) {
    if (!root) return true;
    if (!is_bst(bst[root].lch, pre)) return false;
    if (pre && pre > root) return false;
    pre = root;
    return is_bst(bst[root].rch, pre);
}

bool is_cbt(int root) {
    if (!root) return true;
    queue<int> q;
    int l, r;
    bool leaf = false;
    q.push(root);
    while (!q.empty()) {
        int now = q.front();
        q.pop();
        l = bst[now].lch;
        r = bst[now].rch;
        if ((leaf && (l || r)) || (!l && r)) return false;
        if (l) q.push(l);
        if (r) q.push(r);
        else leaf = true;
    }
    return true;
}

int main(void) {
    scanf("%d%d", &n, &rt);
    while (n--) {
        scanf("%d%d%d", &fa, &lch, &rch);
        bst[fa].lch = lch;
        bst[fa].rch = rch;
    }
    int pre = 0;
    puts(is_bst(rt, pre) ? "true" : "false");
    puts(is_cbt(rt) ? "true" : "false");
    return 0;
}