天梯赛练习 L3-016 二叉搜索树的结构 (30分)

题目分析：

用数型结构先建树，一边输入一边建立，根节点的下标为1，所以左孩子为root*2，右孩子为root*2+1，输入的时候可用cin输入字符串也可用scanf不会超时，判断是否在同一层可以判断两个节点位置以2为底的对数向下取整是否相同得知（log(m)/log(2)为以换底公式实现的求以2为底m的对数）

坑点：

查询的整数可能并不出现在建立的树中（第三个测试数据）

本题代码：（菜）

#include<iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<algorithm>
#include<map>
#include<cmath>
using namespace std;

const int N = 10005;
int vis[N]; 
int tree[N];
map<int, int> mp;
int n, m;

void create(int root, int value){
    if(vis[root] == 0){
        vis[root] = 1;
        tree[root] = value;
        mp[value] = root;        //记录每个value在那个点 
        return;
    }else{
        if(value < tree[root]) create(root*2, value);
        else create(root*2+1, value);
    }
}

bool judge(int x){
    if(mp.find(x) == mp.end()){
        printf("No\n");
        return false;
    }
    return true;
}

void isgen(int x){
    if(!judge(x)) return;
    if(tree[1] == x) printf("Yes\n");
    else printf("No\n");
}

void iscen(int x, int y){
    if(!judge(x) || !judge(y)) return;
    x = mp[x]; y = mp[y];
    if(int(log(x)/log(2)) == int(log(y)/log(2))) printf("Yes\n");
    else printf("No\n");
}

void isleft(int x, int y){
    if(!judge(x) || !judge(y)) return;
    if(mp[y]*2 == mp[x]) printf("Yes\n");
    else printf("No\n"); 
}

void isright(int x, int y){
    if(!judge(x) || !judge(y)) return;
    if(mp[y]*2+1 == mp[x]) printf("Yes\n");
    else printf("No\n");
}

void isborther(int x, int y){
    if(!judge(x) || !judge(y)) return;
    if(mp[x]/2 == mp[y]/2) printf("Yes\n");
    else printf("No\n");
} 

void isparent(int x, int y){
    if(!judge(x) || !judge(y)) return;
    if(mp[y]/2 == mp[x]) printf("Yes\n");
    else printf("No\n");
}

int main(){
    scanf("%d", &n);
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i <= n; i++){
        int x; scanf("%d", &x);
        create(1, x);
    }
    scanf("%d", &m);
    for(int i = 1; i <= m; i++){
        int x;
        cin>>x;
        string s1; cin>>s1;
        if(s1 == "and"){
            int y; cin>>y;
            cin>>s1; cin>>s1;
            if(s1 == "on"){    //判断是否是同层 
                getline(cin, s1);
                iscen(x, y);
            }else{            //判断是否是兄弟 
                getline(cin, s1);
                isborther(x, y);
            }
        }else{
            cin>>s1; cin>>s1;
            if(s1 == "root"){//是否为根 
                isgen(x);
            }else if(s1 == "parent"){//x是否是y的父亲 
                cin>>s1;
                int y; cin>>y;
                isparent(x, y);
            }else if(s1 == "left"){//x是否是y的左孩子 
                cin>>s1; cin>>s1;
                int y; cin>>y;
                isleft(x, y);
            }else{                    //x是否是y的右孩子 
                cin>>s1; cin>>s1;
                int y; cin>>y;
                isright(x, y);
            }
        }
    }
    return 0;
}