试题 I: BST 插入节点问题
时间限制: 1.0s 内存限制: 512.0MB 本题总分:25 分
【问题描述】
给定一棵包含 N 个节点的二叉树,节点编号是 1 ∼ N。其中 i 号节点具有
权值 Wi,并且这些节点的权值恰好形成了一棵排序二叉树 (BST)。
现在给定一个节点编号 K,小明想知道,在这 N 个权值以外,有多少个整
数 X (即 X 不等于任何 Wi ) 满足:给编号为 K 的节点增加一个权值为 X 的子
节点,仍可以得到一棵 BST。
例如在下图中,括号外的数字表示编号、括号内的数字表示权值。即编号
1 ∼ 4 的节点权值依次是 0、10、20、30。
如果 K = 1,那么答案为 0。因为 1 号节点已经有左右子节点,不能再增
加子节点了。
如果 K = 2,那么答案为无穷多。因为任何一个负数都可以作为 2 的左子
节点。
如果 K = 3,那么答案为 9。因为 X = 11, 12, · · · , 19 都可以作为 3 的左子
节点。
【输入格式】
第一行包含 2 个整数 N 和 K。
以下 N 行每行包含 2 个整数,其中第 i 行是编号为 i 的节点的父节点编号
Pi 和权值 Wi 。注意 Pi = 0 表示 i 是根节点。
输入保证是一棵 BST。
【输出格式】
一个整数代表答案。如果答案是无穷多,输出 −1。
【样例输入】
4 3
0 10
1 0
1 20
3 30
【样例输出】
9
【评测用例规模与约定】
对于 60% 的评测用例,1 ≤ K ≤ N ≤ 100,0 ≤ Wi ≤ 200,且 Wi 各不相同。
对于所有评测用例,1 ≤ K ≤ N ≤ 10000,0 ≤ Wi ≤ 100000000,且 Wi 各不 相同。
class TreeNode:
def __init__(self,x):
self.val=x
self.left=None
self.right=None
self.parent=None
N,K=map(int,input().strip().split())
map_=[]
map_node=[]
for i in range(N):
p,w=map(int,input().strip().split())
map_.append([p,w])
map_node.append(TreeNode(w))
if p==0:
root_index=i
for i in range(N):
if map_node[map_[i][0]-1].val<map_node[i].val:
map_node[map_[i][0]-1].right=map_node[i]
map_node[i].parent=map_node[map_[i][0]-1]
else:
map_node[map_[i][0]-1].left=map_node[i]
map_node[i].parent=map_node[map_[i][0]-1]
if map_node[K-1].left and map_node[K-1].right:
print(0)
elif map_node[K-1].right:
if not map_node[K-1].parent:
print(-1)
else:
print(map_node[K-1].val-map_node[K-1].parent.val-1)
elif map_node[K-1].left:
print(-1)
4 3
0 10
1 0
1 20
3 30
9