记录一个菜逼的成长。。
这里建堆要每输入一个值就插入并且向上调整
在堆顶设置一个哨兵会省点操作
还要注意值有负数,在处理字符串的时候要注意。(之前一直不知道哪里有问题,后来才发现负数忘了处理,一直过不了)
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define pb push_back
typedef long long LL;
const int INF = 0x3f3f3f3f;
const int maxn = 2000 + 10;
int heap[maxn],n,m;
string str;
void Insert(int pos,int x)
{
///上滤
heap[pos] = x;
for( int i = pos; heap[i/2] > heap[i] ; i /= 2 ){
swap(heap[i/2],heap[i]);
}
}
bool check1(int a,int b)
{
for( int i = 1; i <= n; i++ )
if((heap[i<<1] == a && heap[i<<1|1] == b) || (heap[i<<1|1] == a && heap[i<<1] == b))return true;
return false;
}
bool check2(int a,int b)
{
for( int i = 1; i <= n; i++ ){
if(heap[i] == a && (heap[i<<1] == b || heap[i<<1|1] == b))return true;
}
return false;
}
int main()
{
cin>>n>>m;
heap[0] = -INF;///哨兵
for( int x,i = 1; i <= n; i++ ){
cin>>x;
Insert(i,x);
}
/*for( int i = 1; i <= n; i++ )
cout<<treap[i]<<' ';
puts("");
*/
int a,b,flag;
while(m--){
cin>>a;getline(cin,str);
flag = 0;
if(str.find("root") != string::npos){
if(a == heap[1])puts("T");
else puts("F");
}
if(str.find("siblings") != string::npos){
b = 0;
for( int i = 0; i < str.size(); i++ ){
if(isdigit(str[i]))b = b * 10 + str[i] - '0';
if(str[i] == '-')flag = 1;
}
if(flag)b = -b;
puts(check1(a,b) ? "T" : "F");
}
if(str.find("parent") != string::npos){
b = 0;
for( int i = 0; i < str.size(); i++ ){
if(isdigit(str[i]))b = b * 10 + str[i] - '0';
if(str[i] == '-')flag = 1;
}
if(flag)b = -b;
puts(check2(a,b)?"T":"F");
}
if(str.find("child") != string::npos){
b = 0;
for( int i = 0; i < str.size(); i++ ){
if(isdigit(str[i]))b = b * 10 + str[i] - '0';
if(str[i] == '-')flag = 1;
}
if(flag)b = -b;
puts(check2(b,a)?"T":"F");
}
}
return 0;
}