蓝桥杯 试题 I: BST 插入节点问题

试题 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 各不
相同。

import java.util.Scanner;

public class BST插入节点问题 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		int[][] node = new int[n + 1][4];
		int max = 0;
		for (int i = 1; i <= n; i++) {
			int p = sc.nextInt();
			int w = sc.nextInt();
			node[i][1] = p;
			node[i][0] = w;
			if(p != 0) {//判断根节点
				if(node[p][0] > node[i][0]) {//判断左节点
					node[p][2] = i;
				}else{//右节点
					node[p][3] = i;
				}
			}
			
		}
		if(node[k][3] != 0 && node[k][2] != 0) {//有左右节点
			System.out.println(0);
			return;
		}else {
			if(node[k][3] == 0 && node[k][2] == 0) {//没有子节点，叶子节点
				System.out.println(-1);
				return;
			}
			if(node[k][2] == 0) {//k没有左节点
				if(node[k][1] == 0) {//k为根节点
					System.out.println(-1);
					return;
				}
				else {
					if(node[node[k][1]][2] == k) {//k为左子树
						System.out.println(-1);
						return;
					}
					else {//k为右子树
						max = node[k][0] - node[node[k][1]][0] - 1;//记录左节点个数
					}
				}
			}
			if(node[k][3] == 0) {//没有右节点
				if(node[k][1] == 0) {
					System.out.println(-1);
					return;
				}
				else {
					if(node[node[k][1]][3] == k) {
						System.out.println(-1);
						return;
					}
					else {
						max = Math.max(max,node[node[k][1]][0] - node[k][0] - 1);

					}
				}
			}
		}
		System.out.println(max);
	}
}