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;
}