7-42 关于堆的判断 (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
树左孩子是2*i,右孩子是2*i+1,则父节点是i/2
#include<bits/stdc++.h>
using namespace std;
const int N=10010;
int h[N];
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>h[i];
for(int i=1;i<=n;i++){//创建最小堆
int k=i;
while(k>1&&h[k]<h[k/2]){//从下标2开始创建
swap(h[k],h[k/2]);
k=k/2;
}
}
while(m--){
int x;string s;
cin>>x;
getchar();
getline(cin,s);
int sum=0,k=0,k1=1;
for(int i=0;s[i];i++)//26 and -12 are siblings
{
if(s[i]>='0'&&s[i]<='9'){
int q=i;
if(s[q-1]=='-')k1=-1;
sum=sum*10+s[i]-'0';
}
}
sum=sum*k1;
if(s.find("root")!=-1){if(x==h[1])cout<<"T"<<endl;else cout<<"F"<<endl;}
else{
int xia1=0,xia2=0;
for(int i=1;i<=n;i++){
if(x==h[i])xia1=i;
if(sum==h[i])xia2=i;
}
if(s.find("siblings")!=-1){
if(h[xia1/2]==h[xia2/2])cout<<"T"<<endl;
else cout<<"F"<<endl;
}
if(s.find("parent")!=-1){
if(x==h[xia2/2])cout<<"T\n";
else cout<<"F\n";
}
if(s.find("child")!=-1){
if(h[xia2]==h[xia1/2])cout<<"T\n";
else cout<<"F\n";
}
}
//cout<<s[9]<<endl;
}
return 0;
}