L2-012. 关于堆的判断

L2-012. 关于堆的判断

题目链接：https://www.patest.cn/contests/gplt/L2-012

终于ac了，简直要哭。题目还是很简单的，不过很多坑：

1.寻找x下标时，有可能返回0，即x是根结点；

2.字符串中字符的位置有可能会因串中的数字长度大小改变而改变（QAQ找了一个小时才发现是在这里）；

3.gets(函数)会读取前一个分隔符（是我基础不好╮(╯▽╰)╭）.

代码如下：

#include<cstdio>
#include<iostream>
using namespace std;
int a[1005];
int location(int key){
int k=0;
while(a[k]!=key)k++;
return k;
}
int transint(char s[]){
if(s[0]=='-'){
int temp=0;
for(int i=1;s[i]!=' '&&s[i]!='\0';i++)
temp=temp*10+s[i]-'0';
return -temp;
}else{
int temp=0;
for(int i=0;s[i]!=' '&&s[i]!='\0';i++)
temp=temp*10+s[i]-'0';
return temp;
}
}
int main(void){
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%d",a+i);
int k=i;
while(k&&a[k]<a[(k-1)/2]){
swap(a[k],a[(k-1)/2]);
k=(k-1)/2;
}
}
while(m--){
bool flag;
char s[50];
int x;
scanf("%d",&x);
gets(s);//由于gets会读取x后的分隔符，所以字符串s实际上是从' '开始的
//一开始用s[8]区分，发现第二种查询会因数字长度改变而改变
if(s[4]=='t'&&s[8]=='r'){//第一种查询
if(a[0]==x)flag=1;
else flag=0;
}else if(s[4]==' '){//第二种查询
int y=transint(&s[5]);
int t=location(x);
if(t){
if(t&1){
if(a[t+1]==y)flag=1;
else flag=0;
}else{
if(a[t-1]==y)flag=1;
else flag=0;
}
}else flag=0;
}else if(s[4]=='t'&&s[8]=='p'){//第三种查询
int y=transint(&s[18]);
int t=location(y);
if(t&&a[(t-1)/2]==x)flag=1;
else flag=0;
}else if(s[4]=='a'){//第四种查询
int y=transint(&s[15]);
int t=location(x);
if(t&&a[(t-1)/2]==y)flag=1;
else flag=0;
}
if(flag)printf("T\n");
else printf("F\n");
}
return 0;
}