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

最新推荐文章于 2023-01-12 10:53:52 发布

原创最新推荐文章于 2023-01-12 10:53:52 发布 · 346 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#堆 #模拟

CCCC 同时被 2 个专栏收录

109 篇文章

订阅专栏

数据结构

13 篇文章

订阅专栏

本文介绍如何通过数组实现小顶堆，并讨论了如何判断堆中元素间的关系，如根节点、兄弟节点、父子节点等。使用C++代码演示了小顶堆的构建过程及各种关系的判断。

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

题目

将一系列给定数字顺序插入一个初始为空的小顶堆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。

输入样例：

5 4
46 23 26 24 10
24 is the root
26 and 23 are siblings
46 is the parent of 23
23 is a child of 10

输出样例：

F
T
F
T

思路

小顶堆的实现+字符串模拟。借此机会学习了一下堆的建立以及姊妹关系、父子关系的判断。

用数组实现小顶堆，每次在末尾添加新节点，并递归的与父节点比较（除2即是父节点）并提上去。

姊妹关系的判断可以从父节点来考虑，节点 $i$ 的子节点是 $2 * i$ 和 $2 * i + 1$ ，所以判断所有的 $\in [1, (cnt - 1) / 2]$ ，是否有 $a_{2 * i} = a_{2 * i + 1}$ 。

父节点和子节点的关系是： $i_{son} = i_{par} * 2$ 或 $i_{son} = i_{par} * 2 + 1$ 。找到父节点或子节点然后判断即可。

字符串模拟的部分就不说辽，硬做就行。开始用java写写炸了，可能是contains函数的问题，重写C++就过了。

代码

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define pb push_back
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 2e5 + 19;
const ll mod = 1e9 + 7;

int heap[N];
int cnt;

void down(int n)
{
    if(n == 1)
        return ;
    int i = n / 2;
    if(heap[i] > heap[n])
    {
        swap(heap[i], heap[n]);
        down(i);
    }
}

void insert(int x)
{
    heap[++cnt] = x;
    down(cnt);
}

char s[N];
int a, b;

bool isSib(int a, int b)
{
    for(int i = 2; i < cnt; i += 2)
    {
        if(heap[i] == a && heap[i + 1] == b)
        {
            return 1;
        }
        if(heap[i] == b && heap[i + 1] == a)
        {
            return 1;
        }
    }
    return 0;
}

bool isPar(int par, int son)
{
    int idx = 0;
    for(int i = 1; i <= cnt; i++)
    {
        if(i * 2 > cnt)
            return 0;
        if(heap[i] == par)
        {
            idx = i;
            break;
        }
    }
    if(heap[idx * 2] == son || heap[idx * 2 + 1] == son)
        return 1;
    return 0;
}

int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 0, x; i < n; i++)
    {
        cin >> x;
        insert(x);
    }
    for(int i = 0; i < m; i++)
    {
        cin >> a;
        cin >> s;
        if(s[0] == 'a')
        {
            cin >> b;
            cin >> s;
            cin >> s;
            if(isSib(a, b))
            {
                cout << "T" << endl;
            }
            else
            {
                cout << "F" << endl;
            }
        }
        else
        {
            cin >> s;
            if(s[0] == 'a')
            {
                cin >> s;
                cin >> s;
                cin >> b;
                if(isPar(b, a))
                {
                    cout << "T" << endl;
                }
                else
                {
                    cout << "F" << endl;
                }
            }
            else
            {
                cin >> s;
                if(s[0] == 'p')
                {
                    cin >> s;
                    cin >> b;
                    if(isPar(a, b))
                    {
                        cout << "T" << endl;
                    }
                    else
                    {
                        cout << "F" << endl;
                    }
                }
                else
                {
                    if(heap[1] == a)
                    {
                        cout << "T" << endl;
                    }
                    else
                    {
                        cout << "F" << endl;
                    }
                }
            }
        }
    }
    return 0;
}