PAT 数据结构 关于堆的判断 （25 分）

 

将一系列给定数字顺序插入一个初始为空的小顶堆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

 

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
vector<int> v;
int n;
void updown(int x){//向上更新，要是小于父亲节点，就和父亲节点交换 
	if(x==1)
	    return ;
	while(x!=1){
		if(v[x]<v[x/2]){
			swap(v[x],v[x/2]);
			x=x/2;
		}//小的就上浮 
		else
		    break;
	}
} 
void IsRoot(int x){
	if(v[1]==x)//根节点就是从1开始 
	    cout<<"T"<<endl;
	else
	    cout<<"F"<<endl;
} 
void IsBorther(int x,int y){
	int ix=0;
	int iy=0;
	for(int i=1;i<=n;i++){
		if(v[i]==x)
		    ix=i;
		if(v[i]==y)
		    iy=i;
	}//记录下标 
	if(ix>iy)
	    swap(ix,iy);//确保iy > ix
	if(ix%2==0&&iy-ix==1)//严谨一点 
	    cout<<"T"<<endl;
	else
	    cout<<"F"<<endl;
}
void FindFather(int x,int y){
	int ix=0;
	int iy=0;
	for(int i=1;i<=n;i++){
		if(v[i]==x)
		    ix=i;
		if(v[i]==y)
		    iy=i;
	}
	if(2*ix==iy||(2*ix)+1==iy)
	    cout<<"T"<<endl;
	else
	    cout<<"F"<<endl;
}
void FindSon(int x,int y){
	int ix=0;
	int iy=0;
	for(int i=1;i<=n;i++){
		if(v[i]==x)
		    ix=i;
		if(v[i]==y)
		    iy=i;
	}
	if(ix==iy*2||ix==2*iy+1)
	    cout<<"T"<<endl;
	else
	    cout<<"F"<<endl;
}
int main(){
	int m;
	int x,y;
	char c[105];
	cin>>n>>m;
	v.resize(n+5);
	for(int i=1;i<=n;i++){
		cin>>v[i];
		updown(i);
	} 
	while(m--){
		cin>>x>>c;
		if(c[0]=='a'){
			cin>>y>>c>>c;
			IsBorther(x,y); 
		}else{
			cin>>c;
			if(c[0]=='a'){
				cin>>c>>c>>y;
				FindSon(x,y);
			}else{
				cin>>c;
				if(c[0]=='r')
				    IsRoot(x);
				else{
					cin>>c>>y;
					FindFather(x,y);
				}
			}
		}
	}
	return 0;
}