PTA-L2-012 关于堆的判断 (25分)
这道题关于堆的问题。
小顶堆:根节点小于或等于所有左右子结点
大顶堆:根节点大于或等于所有左右子结点
这道题构造一个小顶堆。
每读入一个数字就把它加进堆里面。
然后进行堆的调整。比较左右子结点与根节点的大小。进行调整。
把堆建好了,我们需要记录结点和结点所在的位置。用map来记录即可。
让结点作为map的first,位置作为second.
然后就是判断部分了。
有结论:
堆在连续数组的存储中,任意一个结点的父结点是位置/2;
我们可以用上面的结论进而判断。
代码部分:
#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
const int N = 1e3 + 10;
const int M = 55;
const int INF = 1e6 + 10;
typedef long long ll;
int n, m;
int a[N];
int cnt;
map<int, int> mp;
void ins(int x)
{
a[++cnt] = x;
int t = cnt;
while (t > 1 && a[t / 2] > a[t])
{
int k = a[t];
a[t] = a[t / 2];
a[t / 2] = k;
t /= 2;
}
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
int x;
scanf ("%d", &x);
ins(x);
}
for (int i = 1; i <= n; i++)
{
mp[a[i]] = i;
}
int x1, x2;
string s;
while (m--)
{
cin >> x1 >> s;
if (s[0] == 'a')
{
cin >> x2;
cin >> s;
cin >> s;
if (mp[x1] / 2 == mp[x2] / 2)
{
cout << "T\n";
}
else
{
cout << "F\n";
}
}
else
{
cin >> s;
cin >> s;
if (s[0] == 'r')
{
if (mp[x1] != 1)
{
cout << "F\n";
}
else
{
cout << "T\n";
}
}
else if (s[0] == 'p')
{
cin >> s;
cin >> x2;
if (mp[x2] / 2 == mp[x1])
{
cout << "T\n";
}
else
{
cout << "F\n";
}
}
else if (s[0] == 'c')
{
cin >> s;
cin >> x2;
if (mp[x1] / 2 == mp[x2])
{
cout << "T\n";
}
else
{
cout << "F\n";
}
}
}
}
return 0;
}