PAT甲级1135

做出来了,但感觉心好痛。我发现我真是太懒惰了,以为简简单单记个模板,就万事大吉了。

没想到自己的思维被模板固化,就不会根据实际情况作出相应的改变了。模板当然要记忆,

但不可偷懒而导致思维僵化。引以为戒

dfs为判断过程,是在求树高的基础上,增加一个参数,判断是否存在前一个为红色节点并且当前节点是红色的情况

回溯判断当前节点的左右子树的黑高是否相等

#include<cstdio>
#include<algorithm>
#include<cmath>
//#define debug
using namespace std;
const int maxn=40;
struct node{
    int data;
    int lc;
    int rc;
    node(){
        data=0;
        lc=rc=-1;
    }
};
bool cmp(int a,int b)
{
    return abs(a)<abs(b);
}
int n;
int ptr=0;
node Tree[maxn];
int pre[maxn];
int in[maxn];
void create(int &root,int preL,int preR,int inL,int inR)
{
    if(preL>preR)return;
    if(preL>=n)return;
    int roote=pre[preL];
    root=ptr++;
    Tree[root].data=roote;
    int u=0;
    for(u=inL;u<=inR;u++){
        if(in[u]==roote)break;
    }
    int numLeft=u-inL;
    create(Tree[root].lc,preL+1,preL+numLeft,inL,u-1);
    create(Tree[root].rc,preL+numLeft+1,preR,u+1,inR);
}
bool IsRBt=true;
int dfs(int root,int pren)
{
    if(IsRBt==false)return -1;
    if(root==-1)return 1;
    if(pren<0&&Tree[root].data<0){
        IsRBt=false;
        return -1;
    }
    int lh=dfs(Tree[root].lc,Tree[root].data);
    int rh=dfs(Tree[root].rc,Tree[root].data);
    if(rh!=lh){
        IsRBt=false;
        return -1;
    }
    return Tree[root].data>0?max(lh,rh)+1:max(lh,rh);
}
int main()
{
#ifdef debug
    freopen("in.txt","r",stdin);
#endif
    int t;
    node s;
    scanf("%d",&t);
    for(int i=0;i<t;i++){
        scanf("%d",&n);
        fill(Tree,Tree+maxn,s);
        ptr=0;
        IsRBt=true;
        for(int i=0;i<n;i++){
            scanf("%d",&pre[i]);
            in[i]=pre[i];
        }
        sort(in,in+n,cmp);
        int root;
        create(root,0,n-1,0,n-1);
        if(Tree[root].data<0)IsRBt=false;
        else
            dfs(root,Tree[root].data);
        if(IsRBt)printf("Yes\n");
        else printf("No\n");
    }
#ifdef debug
    getchar();
#endif
}

 



 

posted @ 2017-09-24 16:02  MalcolmMeng  阅读(146)  评论(0编辑  收藏  举报