pat甲级真题分类-树

Is It A Red-Black Tree

/*
判断一个树是否为红黑树：
1.根节点是黑色
2.节点是红色，左右儿子都是黑色
3.从一个节点出发到叶子结点，其路径上黑色点的数量是相同的

*/
#include<iostream>
#include<unordered_map>
#include<cmath>
#include<algorithm>
using namespace std;

const int maxn=33;

int T;

bool ans;

int pre[maxn],in[maxn];

unordered_map<int,int> pos;

int build(int il,int ir,int prel,int prer,int &sum)
{
    int root=pre[prel];
    int k=pos[abs(root)];

    if(k<il || k>ir)
    {
        ans=false;
         return 0;
    }

    int left=0,right=0,ls=0,rs=0;

    if(il<k)
        left=build(il,k-1,prel+1,prel+k-il,ls);

    if(k<ir)
        right=build(k+1,ir,prel+1+k-il,prer,rs);

    if(ls!=rs)
        ans=false;

    sum=ls;

    if(root<0)
    {
        if(left<0 || right<0)
        {
            ans=false;
        }
    }
    else
        sum++;
    return root;

}

int main()
{
    cin >> T;
    while(T--)
    {
        int n;
        cin >> n;
        for(int i=0; i<n; i++)
        {
            cin >> pre[i];
            in[i]=abs(pre[i]);
        }
        sort(in,in+n);

        pos.clear();

        ans=true;

        for(int i=0; i<n; i++)
            pos[in[i]]=i;
        int sum;
        int root=build(0,n-1,0,n-1,sum);

        if(root<0)
            ans=false;

        if(ans)
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }
    return 0;
}

Postfix Expression

#include<iostream>
using namespace std;

const int maxn=22;

struct node
{
    string data;
    int left=-1,right=-1;
}Node[maxn];

bool isroot[maxn];

void postorder(int root)
{
    if(root==-1)
        return ;
    else if(Node[root].left==-1 || Node[root].right==-1)        //通过观察发现，只要是左子树或者右子树残缺的，我们是前序遍历，对于剩下的情况都是后序遍历
    {
        printf("(%s",Node[root].data.c_str());
        postorder(Node[root].left);
        postorder(Node[root].right);
        cout << ")";
    }
    else
    {
        cout << "(";
        postorder(Node[root].left);
        postorder(Node[root].right);
        printf("%s)",Node[root].data.c_str());
    }
}

int main()
{
    int n,root;
    cin >> n;
    for(int i=1; i<=n; i++)
    {
        cin >> Node[i].data >> Node[i].left >> Node[i].right;
        isroot[Node[i].left]=true;
        isroot[Node[i].right]=true;
    }

    for(int i=1; i<=n; i++)
        if(isroot[i]==false)
            root=i;

    postorder(root);

    return 0;
}