将一系列给定数字顺序插入一个初始为空的小顶堆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”。
分析:
1、边插入边调整。
2、写法问题,注意插入负数的符号处理。
3、堆是完全二叉树,先插入到最后,再向上调整。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1000 + 10;
vector<string> v;
map<int, int> mp;
int a[MAXN];
int change(string s){
int st = 0;
if(s[0] == '-') ++st;
int sum = 0;
for(int i = st; i < s.size(); ++i){
sum = sum * 10 + s[i] - '0';
}
if(st == 1) return -sum;
return sum;
}
int main(){
int N, M;
scanf("%d%d", &N, &M);
int cnt = 0;
while(N--){
++cnt;
scanf("%d", &a[cnt]);
int tmp = cnt;
while(tmp != 1){
if(a[tmp] < a[tmp / 2]){
swap(a[tmp], a[tmp / 2]);
tmp /= 2;
}
else break;
}
}
for(int i = 1; i <= cnt; ++i){
mp[a[i]] = i;
}
getchar();
string s;
while(M--){
v.clear();
getline(cin, s);
stringstream ss(s);
string x;
int id;
while(ss >> x){
v.push_back(x);
if(x == "root") id = 1;
else if(x == "siblings") id = 2;
else if(x == "parent") id = 3;
else if(x == "child") id = 4;
}
int sum1 = change(v[0]);
if(id == 1){
if(a[1] == sum1) printf("T\n");
else printf("F\n");
}
else if(id == 2){
int sum2 = change(v[2]);
if(mp[sum1] / 2 == mp[sum2] / 2) printf("T\n");
else printf("F\n");
}
else if(id == 3){
int sum2 = change(v[5]);
if(mp[sum2] / 2 == mp[sum1]) printf("T\n");
else printf("F\n");
}
else{
int sum2 = change(v[5]);
if(mp[sum1] / 2 == mp[sum2]) printf("T\n");
else printf("F\n");
}
}
return 0;
}