PTA-L2-012 关于堆的判断 (25分)

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;
}