PAT 甲级 1135 Is It A Red-Black Tree (30 分) 指针建树+dfs

最新推荐文章于 2021-09-06 13:43:51 发布

Joey丶sunk

最新推荐文章于 2021-09-06 13:43:51 发布

阅读量190

点赞数

分类专栏： PAT甲级

本文链接：https://blog.csdn.net/xiang_6/article/details/100128763

版权

PAT甲级专栏收录该内容

48 篇文章 0 订阅

订阅专栏

这是一道关于红黑树的编程题目，题目简化了红黑树的表示，所有叶子节点都是NULL。在解决过程中，主要涉及指针构建树结构和深度优先搜索（DFS）的先序遍历应用。

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

马一下，，，这个题真滴蠢

ps：红黑树的叶子结点省略了，全是NULL，先序遍历的时候不会输出

#include<bits/stdc++.h>
#include<cstring>
#define FI first
#define SE second

using namespace std;
typedef long long ll;
typedef pair<string, string> P;
const int maxn = 100 + 7;
int n;
set<int> st;
int arr[maxn];
struct node {
    int v;
    node *l, *r;
};
node *build(node *root, int v) {
    if(root == NULL) {
        root = new node();
        root->v = v;
        root->l = root->r = NULL;
    }
    else if(abs(v) <= abs(root->v)) {
        root->l = build(root->l, v);
    }
    else {
        root->r = build(root->r, v);
    }
    return root;
}
bool judge1(node *root) {
    if (root == NULL) return true;
    if (root->v < 0) {
        if (root->l != NULL && root->l->v < 0) return false;
        if (root->r != NULL && root->r->v < 0) return false;
    }
    return judge1(root->l) && judge1(root->r);
}
void dfs(node *root, int cnt) {
    if(root == NULL) {
        st.insert(cnt);
        return;
    }
    int t = 0;
    if(root->v > 0) t = 1;
    dfs(root->l, cnt+t);
    dfs(root->r, cnt+t);
}
bool judge2(node *root) {
    if (root == NULL) return true;
    st.clear();
    dfs(root, 0);
    if(st.size() != 1) return false;
    return judge2(root->l) && judge2(root->r);
}
int main() {
    int T; cin >> T;
    while(T--) {
        cin >> n;
        node *root = NULL;
        for(int i = 0; i < n; ++i) {
            cin >> arr[i];
            root = build(root, arr[i]);
        }
        if(arr[0] < 0 || judge1(root) == false || judge2(root) == false) {
            puts("No");
        }
        else {
            puts("Yes");
        }
    }

    return 0;
}