对于堆的建立方法各不相同, 我已开始学的时候是先将序列输入然后从倒数第一层开始adjustheap() 然后将每一个父节点都调整好,最后便是堆, 但是这道题由于建堆不对,显然同一序列的数建立堆是不唯一的, 所以我们在建立堆的时候要注意题目的要求。
这道题是一边插入一边建立堆, 我们插入一个元素, 然后把他放到适合的位置,即询问其父节点是否小于该数,若小于将父节点放到该位置,进行若干次操作后把其放到正确位置。
正确性: 首先我们保证每次插入完都是堆,所以我们在一个堆中插入一个元素与其父节点比较,小于则进行交换,若干次操作后,能将该元素放到正确位置上。
#include <bits/stdc++.h>
#define rep(i, j, k) for(int i = j; i <= k; i ++)
#define per(i, j, k) for(int i = j; i >= k; i --)
#define f first
#define s second
#define pb(x) push_back(x)
#define ll long long
#define lc p<<1
#define rc p<<1|1
using namespace std;
const int N = 1e4;
int heap[N], len;
void PushUp(int i){
if (i == 1) return;
while (i != 1){
if (heap[i] < heap[i / 2]) swap(heap[i], heap[i / 2]), i /= 2;
else break;
}
}
bool judge(int x, int y, int p){
if (heap[p] == x && p <= len){
if (p == 1) return false;
if (p % 2 == 0 && p < len && heap[p + 1] == y) return true;
if (p % 2 != 0 && p <= len && heap[p - 1] == y) return true;
return false;
}
else if (p > len) return false;
bool cnt = false;
cnt |= judge(x, y, lc);
cnt |= judge(x, y, rc);
return cnt;
}
bool judge2(int x, int y, int p){
if (heap[p] == y && p <= len){
if (heap[p / 2] == x) return true;
return false;
}
else if (p > len) return false;
bool cnt = false;
cnt |= judge2(x, y, lc);
cnt |= judge2(x, y, rc);
return cnt;
}
int main()
{
int n, m;
string a;
cin >> n >> m;
rep (i, 1, n){
cin >> heap[i];
PushUp(i);
}
len = n;
while (m --){
int x, y;
bool flag = false;
string t;
cin >> x >> t;
if (t == "and"){
cin >> y;
getline(cin, t);
if (judge(x, y, 1)) flag = true;
}
else{
cin >> t;
if (t == "the"){
cin >> t;
if (t == "root"){
if (heap[1] == x) flag = true;
}
else if(t == "parent"){
cin >> t >> y;
if (judge2(x, y, 1)) flag = true;
}
}
else{
cin >> t; cin >> t;
cin >> y;
if (judge2(y, x, 1)) flag = true;
}
}
if (flag) cout << "T";
else cout << "F";
if (m) cout << "\n";
}
}