题目描述
有 (n≤7)个国家参加世界杯决赛圈且进入淘汰赛环节。已经知道各个国家的能力值,且都不相等。能力值高的国家和能力值低的国家踢比赛时高者获胜。1 号国家和 2 号国家踢一场比赛,胜者晋级。3 号国家和 4 号国家也踢一场,胜者晋级……晋级后的国家用相同的方法继续完成赛程,直到决出冠军。给出各个国家的能力值,请问亚军是哪个国家?
输入格式
第一行一个整数 n,表示一共 个国家参赛。
第二行 个整数,第i个整数表示编号为 i的国家的能力值(1≤i≤
)。
数据保证不存在平局。
输出格式
仅一个整数,表示亚军国家的编号。
输入输出样例
输入 #1
3
4 2 3 1 10 5 9 7
输出 #1
1
解题思路
本题的关键在于线段树的构建,线段树本质上是一颗二叉树,其中每个节点代表一个区间,对于给定的区间[L , R],一般情况下如果父节点代表区间[L , R],那么它的两个子节点分别代表区间[L , (L+R)/2]和[(L+R)/2 +1 , R],叶子节点代表区间中的单个元素。
本题中的例#1,对于8个国家,要确定谁是亚军。我们需要将问题一步步拆解:
如上图,要确定8个国家谁是亚军,首先将1~8的国家分为1~4和5~8的国家,然后1~4的国家和5~8的国家再分,直到分为单个国家为止,对于单个国家自己就是冠军,对于两个国家谁的能力大谁就是冠军,依次再向上推,直到确定出最终的冠军。图中蓝色部分即为,线段树的所有节点,满足如果父节点代表区间[L , R],那么它的两个子节点分别代表区间[L , (L+R)/2]和[(L+R)/2 +1 , R]的条件。画出最终的线段树如下:
可以看到,这颗树也是一颗满二叉树,对于满二叉树,若父节点为n则左右子节点分别为2n和2n + 1,则亚军为tree[2] > tree[3] ? tree[3] : tree[2], 从图中可得亚军为“4”对应的代表区间为1~1,故最终输出结果为1。
源代码:
import java.util.Scanner;
class Tree {
int power, id;
Tree(int power, int id) {
this.power = power;
this.id = id;
}
}
public class Main {
static Tree[] a = new Tree[150];
static Tree[] tree = new Tree[600];
static Tree maxt(Tree a, Tree b) {
return a.power > b.power ? a : b;
}
static Tree mint(Tree a, Tree b) {
return a.power < b.power ? a : b;
}
static void build(int node, int start, int end) {
if (start == end) {
tree[node] = a[start];
return;
}
int l = node * 2;
int r = node * 2 + 1;
int mid = (start + end) / 2;
build(l, start, mid);
build(r, mid + 1, end);
tree[node] = maxt(tree[l], tree[r]);
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n; // 输入
n = scanner.nextInt();
for (int i = 1; i <= (1 << n); i++) {
int power = scanner.nextInt();
a[i] = new Tree(power, i);
}
build(1, 1, (1 << n));
System.out.println(mint(tree[2], tree[3]).id);
}
}
扩展:
本题要求的是亚军,即要求生成线段树时取区间中的最大值,最终可得到结果,只要掌握线段树的生成,也可以根据区间中的最小值,以及区间中所有值的和来创建树,以应对不同题目。