PAT A1135 Is It A Red-Black Tree (30 分) 二叉搜索树

   题目大意：给出红黑树满足的5个性质，然后给出先序序列，判断是否是红黑树。

   题目给出的红黑树的5个性质如下：

• (1) Every node is either red or black. 每个节点要么是红色，要么是黑色。
• (2) The root is black. 根节点是黑色。
• (3) Every leaf (NULL) is black. 每个叶子节点（即NULL节点）是黑色。
• (4) If a node is red, then both its children are black. 红色节点的左右孩子都是黑色。
• (5) For each node, all simple paths from the node to descendant leaves contain the same number of black nodes. 对每个节点而言，到达它的每个叶子节点的简单路径上，经过的黑色节点总数是一致的。

       这5个性质里，第3个和第5个我一时没看懂，还是wiki了一下才明白的。红黑树的定义里，叶子节点是NULL节点，而不是通常所说的左右孩子为空的节点是NULL节点，也就是说每个左右孩子为空的节点，实际上左右孩子都是叶子节点。然后，性质5是说每个节点到达叶子节点(NULL）节点的简单路径上经过的黑色节点总数相等。

       因为红黑树也是BST，满足中序序列有序，然后根据先序和中序序列构建树。之后就是要检验这棵树是否满足5个性质，性质2 3 4都容易检验。

        问题在于性质5，因为是二叉树，所以这个性质实际上是在说，每个节点的左右子树的黑色节点数量是相等的。那么使用dfs，从根节点开始，对每一个节点的左右子树的黑色节点数量做统计，如果有一个节点不满足左右子树黑色节点数量相等的条件，就不是红黑树。递归的边界是叶子节点，应当返回数量是1。在dfs的过程中，可以顺便检验性质4，即遇到红色节点时检查它的左右孩子是否是黑色的。

        非常奇怪的一点是，如果用 map<int, int> 存储中序遍历的结果，测试点1会内存超限。从给定的数据量范围 <= 30来看，似乎不应该出现这种情况。

AC代码：

#include <bits/stdc++.h>

using namespace std;

struct Node
{
    int val;
    Node* left;
    Node* right;
    Node(int val):val(val), left(NULL), right(NULL){};
};

void createFromPreIn(Node* &node, int prel, int prer, int inl, int inr, vector<int> &pre, vector<int>& in)
{
    if(prel > prer) return;
    if(!node) node = new Node(pre[prel]);
    int index;
    for(index = inl; index <= inr; index ++)
    {
        if(in[index] == abs(pre[prel])) break;
    }
    int leftNum = index - inl;
    createFromPreIn(node->left, prel + 1, prel + leftNum, inl, index - 1, pre, in);
    createFromPreIn(node->right, prel + leftNum + 1, prer, index + 1, inr, pre, in);
}

bool isBlack(Node* node)
{
    return (node == NULL || node->val > 0);
}

int dfs(Node* node, bool &flag)
{
    if(!flag) return 0;
    if(!node) return 1;
    if(!isBlack(node) && (!isBlack(node->left) || !isBlack(node->right))) flag = false;
    int leftBlackCnt = dfs(node->left, flag);
    int rightBlackCnt = dfs(node->right, flag);
    if(leftBlackCnt != rightBlackCnt) flag = false;
    return leftBlackCnt + (isBlack(node) ? 1 : 0);
}

int main()
{
    int K;
    scanf("%d", &K);
    for (int query = 0; query < K; ++query)
    {
        int N;
        scanf("%d", &N);
        vector<int> pre(N);
        vector<int> absPre(N);
        vector<int> in(N);
        for (int i = 0; i < N; ++i)
        {
            scanf("%d", &pre[i]);
            in[i] = abs(pre[i]);
        }
        sort(in.begin(), in.end());
        Node* root = NULL;
        createFromPreIn(root, 0, N-1, 0, N-1, pre, in);
        bool flag = isBlack(root);
        dfs(root, flag);
        if(flag) printf("Yes\n");
        else printf("No\n");
    }
}