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 }