L2-012 关于堆的判断

L2-012 关于堆的判断
将一系列给定数字顺序插入一个初始为空的小顶堆H[]。随后判断一系列相关命题是否为真。命题分下列几种：

“x is the root”：x是根结点；
“x and y are siblings”：x和y是兄弟结点；
“x is the parent of y”：x是y的父结点；
“x is a child of y”：x是y的一个子结点。
输入格式：

每组测试第1行包含2个正整数N（<= 1000）和M（<= 20），分别是插入元素的个数、以及需要判断的命题数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。之后M行，每行给出一个命题。题目保证命题中的结点键值都是存在的。

输出格式：

对输入的每个命题，如果其为真，则在一行中输出“T”，否则输出“F”。

输入样例：

5 4
46 23 26 24 10
24 is the root
26 and 23 are siblings
46 is the parent of 23
23 is a child of 10
输出样例：

F
T
F
T

解题思路：
小堆顶的操作，因为要按顺序插入，所以要从下往上调整。自己写的总是只能得1分，无奈使用柳神博客的方法。

#include<cstdio> 
#include<vector>
#include<cstring>
using namespace std;
vector<int>heap;
int n,m;
void upadjust(int i){
    if(i==1) return ;
    while(i!=1){
        if(heap[i/2]>heap[i]){
            swap(heap[i/2],heap[i]);
            i=i/2;
        }else{
            break;
        }
    }
}
void judge1(int a){
    if(heap[1]==a)  printf("T\n");
    else printf("F\n");
}
void judge2(int a,int b){
    int indexa=0,indexb=0;
    for(int i=1;i<=n;i++){      //在数组中找到a,b的位置 
        if(heap[i]==a) indexa=i;
        if(heap[i]==b) indexb=i;
    }
    if(indexa>indexb)   swap(indexa,indexb);
    if(indexa%2==0&&indexb-indexa==1) printf("T\n");
    else printf("F\n");
}
void judge3(int a,int b){
    int indexa=0,indexb=0;
    for(int i=1;i<=n;i++){
        if(heap[i]==a) indexa=i;    
        if(heap[i]==b) indexb=i;
    }
    if((indexa*2)==indexb||(indexa*2)+1==indexb)    printf("T\n");
    else printf("F\n");
}
void judge4(int a,int b){
    int indexa=0,indexb=0;
    for(int i=1;i<=n;i++){
        if(heap[i]==a) indexa=i;    
        if(heap[i]==b) indexb=i;
    }
    if(indexb*2==indexa||indexb*2+1==indexa)    printf("T\n");
    else printf("F\n");
}
int main(){
    scanf("%d%d",&n,&m);
    heap.resize(n+1) ;  //没有这一句就段错误 
    for(int i=1;i<=n;i++){
        scanf("%d",&heap[i]);
        upadjust(i) ;
    }
    for(int i=0;i<m;i++){
        int a,b;
        char c[110];
        scanf("%d%s",&a,c)  ;
        if(strcmp(c,"and")==0){
            scanf("%d%s%s",&b,c,c);
            judge2(a,b);
        }else{
            scanf("%s",c);
            if(strcmp(c,"a")==0){
                scanf("%s%s%d",c,c,&b);
                judge4(a,b);
            }else{
                scanf("%s",c);
                if(strcmp(c,"root")==0){
                    judge1(a);
                }else{
                    scanf("%s%d",c,&b);
                    judge3(a,b);
                }
            }
        }
    }
    return 0;
}