2017年团体程序设计天梯赛-总决赛 L3-1. 二叉搜索树的结构

#include<bits/stdc++.h>
using namespace std;
int n,t,cnt,f[1024],a,b,fa,fb;
char s[1024];
map<int,int>mp;
struct node
{
int n;
int d;
int l,r;
} p[1024];

void build(int now,int num)
{
if(num<p[now].n)
{
if(p[now].l==-1)
{
p[now].l=cnt;
p[cnt].n=num;
p[cnt].l=p[cnt].r=-1;
cnt++;
}
else build(p[now].l,num);
}
else
{
if(p[now].r==-1)
{
p[now].r=cnt;
p[cnt].n=num;
p[cnt].l=p[cnt].r=-1;
cnt++;
}
else build(p[now].r,num);
}
}

void dfs(int x,int d)
{
mp[p[x].n]=x;
p[x].d=d;
if(p[x].l!=-1)
{
f[p[x].l]=x;
dfs(p[x].l,d+1);
}
if(p[x].r!=-1)
{
f[p[x].r]=x;
dfs(p[x].r,d+1);
}
}

int get()
{
a=b=-1;
int len=strlen(s);
int now=-1,f=1,h1=-1,h2=-1;
for(int i=0; i<len; i++)
{
if(s[i]=='-')
{
f=-1;
}
else if(s[i]>='0'&&s[i]<='9')
{
if(now==-1) now=0;
now=now*10+s[i]-'0';
h2=1;
}
else if(h2!=-1)
{
if(h1==-1) a=f*now;
else b=f*now;
now=-1;
f=1;
h1=1;
h2=-1;
}
}
if(h2!=-1)
{
if(h1==-1) a=f*now;
else b=f*now;
now=-1;
}
for(int i=0; i+2<len; i++)
{
if(s[i]=='r'&&s[i+1]=='o') return 1;
if(s[i]=='s'&&s[i+1]=='i') return 2;
if(s[i]=='p'&&s[i+1]=='a') return 3;
if(s[i]=='e'&&s[i+1]=='f') return 4;
if(s[i]=='r'&&s[i+1]=='i') return 5;
if(s[i]=='s'&&s[i+1]=='a') return 6;
}

}

int main()
{
mp.clear();
cnt=1;
cin>>n;
for(int i=0; i<n; i++)
{
scanf("%d",&t);
if(i==0)
{
p[cnt].n=t;
p[cnt].l=p[cnt].r=-1;
cnt++;
}
else build(1,t);
}
f[1]=-1;
dfs(1,1);
cin>>n;
getchar();
for(int i=0; i<n; i++)
{
gets(s);
int k=get();
int t=0;
fa=mp[a];
fb=mp[b];
//printf("%d %d %d %d\n",a,b,fa,fb);
if(mp[a]==0)
t=0;
else if(k==1)
{
if(f[fa]==-1) t=1;
}
else if(mp[b]==0)
{
t=0;
}
else if(k==2)
{
if(f[fa]==f[fb]) t=1;
}
else if(k==3)
{
if(f[fb]==fa) t=1;
}
else if(k==4)
{
if(p[fb].l==fa) t=1;
}
else if(k==5)
{
if(p[fb].r==fa) t=1;
}
else if(k==6)
{
if(p[fa].d==p[fb].d) t=1;
}
if(t==1) puts("Yes");
else puts("No");
}
return 0;
}