L2-012 关于堆的判断(Python3)

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

提交结果：

代码：

def up(u):
    while (u > 1 and (h[u] < h[int(u / 2)])):
        h[u], h[int(u / 2)] = h[int(u / 2)], h[u]
        u = int(u / 2)


n, m = list(map(int, input().split()))
h = list(map(int, input().split()))
h.insert(0, 0)
pos = {}
for i in range(1, n + 1):
    up(i)
for i in range(1, n + 1):
    pos[h[i]] = i
for i in range(m):
    data1 = input().split()
    x = int(data1[0])
    posx = pos[x]
    if data1[1] == "is":
        if data1[2] == "a":
            y = int(data1[-1])
            posy = pos[y]
            if int(posx / 2) == posy:
                print("T")
            else:
                print("F")
        else:
            if data1[3] == "root":
                if pos[x] == 1:
                    print("T")
                else:
                    print("F")
            else:
                y = int(data1[-1])
                posy = pos[y]
                if int(pos[y] / 2) == pos[x]:
                    print("T")
                else:
                    print("F")
    else:
        y = int(data1[2])
        posy = pos[y]
        if int(posx / 2) == int(posy / 2):
            print("T")
        else:
            print("F")